直接上效果图如下:
代码演示和demo仓库看b站视频:
代码展示:
数据集图片放置在data文件夹下,大家可以根据自己需要比如识别其他物体,只需要模仿data文件夹下的文件命名放入图片即可运行训练模型了。
编辑
运行01数据集文本生成制作.py可以将data文件夹下的图片保存成txt格式
运行02train.py可以将txt中记录的数据读取并训练模型
训练可以使用的卷积神经网络包括: efficientnet、Alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNext、ShuffleNet、Swin_transformer、VGG等。