本期给大家介绍的是含lenet5等多种卷积神经网络识别中文汉字,效果图如下:
这是运行python 03pyqt.py的可视化界面,通过在可视化的pyqt界面画板上画出我们需要识别的汉字,然后点击识别,就会在下方识别出结果。
代码下载和视频演示地址:
106python语言含lenet5等多种卷积神经网络中文汉字识别_哔哩哔哩_bilibili
代码整体是非常简便的,总共三个py部分和一个数据集在data文件夹下。
运行python 01数据集文本生成制作.py 会在logs文件夹下生成2个txt文本,分别存放了汉字数据集图片的路径和对应的标签。
运行python 02train.py就会训练这个txt文本里面的数据,并将训练的模型与验证集里面的数据进行验证。最后模型也是保存在logs文件夹下。
训练完成之后还会在result文件夹下,生成评价指标图。
除了lenet的CNN外,还提供了vgg、resnet、mobilenet、googlenet、densenet等,进行任意的切换。
最后运行python 03pyqt.py即可加载训练好的模型,对画板上面的汉字进行识别。