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

萧鼎

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

yolov5模型训练(自制数据集)

这些东西是我从什么都不知道 熬了十天晚上左右 摸打滚爬于csdn、github总结出来的 在训练过程中遇到什么问题可以在评论区留言或者私信 看到会一一回复 呜呜呜/(ㄒoㄒ)/~~ 1.制作数据集 1.1 lableimg使用&#xff08