直接上效果图:
通过点击按钮可以实现在画板上写汉字识别和加载图片识别两个功能。
视频演示和demo仓库地址在b站视频001期:
到此一游7758258的个人空间-到此一游7758258个人主页-哔哩哔哩视频
所有代码展示:
十分的简洁,主要是01,02,03.py文件
运行01dataset.py可以将data文件下的图片数据集保存成txt格式记录。
运行02train.py可以读取txt记录的图片数据进行训练,训练的模型保存在本地,其中提供了10多种的模型可以任意的切换。包括:efficientnet、Alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNext、ShuffleNet、Swin_transformer、VGG等。
训练结束后保存评价指标图在result文件下:
最后运行03pyqt.py可以展示一个可视化的交互界面,通过点击按钮来识别,这里弹出的界面上提供了第一个按钮为在画板上控制鼠标写出汉字识别。
第二个按钮为加载汉字图片进行识别。