yolo训练模型

前期准备:
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文件夹下打开终端。
yolo训练模型_第1张图片

一、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

训练模型效果如下:
yolo训练模型_第2张图片相应的视频效果如下:

你可能感兴趣的:(目标检测,深度学习,计算机视觉)