本期给大家介绍的是含resnet等多个模型的手写整句或单个中文汉字识别,效果图如下:
这是运行python 03pyqt.py的可视化界面,通过在画板上用鼠标写出中文汉字,再点识别按钮就会识别结果在下方显示。
代码下载和视频演示地址:
108含resnet等多个模型的手写整句或单个中文汉字识别_哔哩哔哩_bilibili
代码整体是非常简便的,总共三个py部分和一个数据集在data文件夹下。
运行01数据集文本生成制作.py 会将data文件夹下的手写汉字图片数据集保存txt格式,里面记录了图片路径和对应的标签。
运行02train.py可以对模型进行训练
训练好的模型同样保存在logs文件夹下。同时生成评价指标图,f1-score,精确度和召回率。
同时提供了除resnet外,还有多种卷积神经网络供任意切换,包括VGG,Swin transformer,ShuffleNet,ResNeXt,Mobilenet,GoogleNet,DLA,DenseNet,AlexNet等等。