yolo3-pytorch训练摩托车模型并且识别

1.使用cuda10.0环境，检测有无cuda10.0环境，使用代码nvcc -V，出现以下图情况则为环境配置已好

2.准备下载yolo3-pytorch框架：https://download.csdn.net/download/liaoqingjian/74061946

3.下载所需要的摩托车数据集：https://download.csdn.net/download/liaoqingjian/74020603
4.将下载好的摩托车xml文件和图片依次放入以下Annotations和JPEGImages两个文件夹

5.修改utils中config.py文件中的类别数为1，即classes=1

6.修改主目录中的voc_annotation.py文件中的classes为motorcycle，如果图片形式为.png格式则需要修改成后缀png，如果图片形式为.jpg格式则需要修改成后缀jpg

7.先右键点击voc2yolo3.py运行

8.然后右键点击voc_annotation.py运行

9.最后运行train.py文件开始训练

10.训练中显示

11.最后训练识别效果展示

版权声明：本文为CSDN博主「萧鼎」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/liaoqingjian/article/details/122322378