110基于HWDB数据集识别多行文字含Mobilenet等多个模型

本期给大家介绍的是基于HWDB数据集识别多行文字含Mobilenet等多个模型，效果图如下：

这是运行python 03pyqt.py的可视化界面，通过加载本地的图片，会将图片中的汉字检测出来。

代码下载和视频演示地址：

110基于HWDB数据集识别多行文字含Mobilenet等多个模型_哔哩哔哩_bilibili

代码整体是非常简便的，总共三个py部分和一个数据集在data文件夹下。

运行01数据集文本生成制作.py 会将data文件夹下的HWDB图片数据集保存txt格式，里面记录了图片路径和对应的标签。

运行02train.py可以对模型进行训练

训练好的模型同样保存在logs文件夹下。同时生成评价指标图，f1-score,精确度和召回率。

同时提供了除Mobilenet外，还有多种卷积神经网络供任意切换，包括VGG，Swin transformer,resnet,ResNeXt,Shufflenet,GoogleNet,DLA,DenseNet,AlexNet等等。

最后运行python 03pyqt.py即可加载训练好的模型，对加载的图片汉字进行识别。