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

1.使用cuda10.0环境,检测有无cuda10.0环境,使用代码nvcc -V,出现以下图情况则为环境配置已好
yolo3-pytorch训练摩托车模型并且识别_第1张图片
2.准备下载yolo3-pytorch框架:https://download.csdn.net/download/liaoqingjian/74061946

3.下载所需要的摩托车数据集:https://download.csdn.net/download/liaoqingjian/74020603
4.将下载好的摩托车xml文件和图片依次放入以下Annotations和JPEGImages两个文件夹
yolo3-pytorch训练摩托车模型并且识别_第2张图片
5.修改utils中config.py文件中的类别数为1,即classes=1
yolo3-pytorch训练摩托车模型并且识别_第3张图片
6.修改主目录中的voc_annotation.py文件中的classes为motorcycle,如果图片形式为.png格式则需要修改成后缀png,如果图片形式为.jpg格式则需要修改成后缀jpg
yolo3-pytorch训练摩托车模型并且识别_第4张图片
yolo3-pytorch训练摩托车模型并且识别_第5张图片
yolo3-pytorch训练摩托车模型并且识别_第6张图片

7.先右键点击voc2yolo3.py运行
yolo3-pytorch训练摩托车模型并且识别_第7张图片
8.然后右键点击voc_annotation.py运行
yolo3-pytorch训练摩托车模型并且识别_第8张图片
9.最后运行train.py文件开始训练
yolo3-pytorch训练摩托车模型并且识别_第9张图片
10.训练中显示
yolo3-pytorch训练摩托车模型并且识别_第10张图片
11.最后训练识别效果展示
yolo3-pytorch训练摩托车模型并且识别_第11张图片
yolo3-pytorch训练摩托车模型并且识别_第12张图片
yolo3-pytorch训练摩托车模型并且识别_第13张图片
yolo3-pytorch训练摩托车模型并且识别_第14张图片
yolo3-pytorch训练摩托车模型并且识别_第15张图片

你可能感兴趣的:(人工智能算法,pytorch,深度学习,python)