yolo训练模型

unbekannten

已于 2022-03-14 21:04:46 修改

阅读量5.1k

点赞数 1

文章标签：目标检测深度学习计算机视觉

于 2022-03-14 20:48:53 首次发布

本文链接：https://blog.csdn.net/unbekannten/article/details/123487518

版权

前期准备：
1.数据集road：https://download.csdn.net/download/unbekannten/84852589
2.数据集处理脚本文件train.py：https://download.csdn.net/download/unbekannten/84852869
3.yolo各版本模型（包括.cfg+train.names+预训练模型）：https://download.csdn.net/download/unbekannten/84856848
4.自己训练出来的模型，可用于跑视频进行测试

下图是目录文件放置位置，yolo文件夹下有darknet和dataSet，在dataSet文件夹下打开终端。
在这里插入图片描述

一、yolofastest训练

python3 train.py --yolo=../darknet --dataset=./road
sudo ../darknet/darknet partial ./yolo-fastest-xl.cfg ./yolo-fastest-xl.weights  yolo-fastest-xl.conv29 29
sudo ../darknet/darknet detector train ./temp2Train/train.data ./yolo-fastest-xl.cfg ./yolo-fastest-xl.conv.29 -map

训练模型效果如下：
在这里插入图片描述相应的视频效果如下：