制作和训练自己的yolov5模型

一，制作训练集
工具:labelimg 下载https://github.com/tzutalin/labelImg
训练集格式
image-
train(训练图片）
val(验证图片）
label-
train（训练标签）
val(验证标签）

open dir 打开要标注图片的文件夹
change save dir选择标签保存的文件夹
greate 框选样本
save 保存或勾选view下第一个自动保存
需要的环境
克隆yolov5项目 网址: https://github.com/ultralytics/yolov5

下载预训练权重文件yolov5s.pt yolov5m.pt yolov5l.pt
yolov5x.pt 并放置在weights文件夹下

在dataset.py和yaml文件下根据自己的需要修改类别class=[‘fire’,‘smoke’]和num_class

在utils/wandb_logging/wandb_utils.py中修改项目名称
训练模型

python3 train.py --data data/mydate.yaml --cfg models/yolov5s.yaml --weights
weights/yolov5s.pt --batch-size 64 --img120 Using CUDA device0 _CudaDeviceProperties(nano=‘NVIDIANTegra X2’,total_memory=7858MB’)
在runs下查看训练结果

找到预训练权重best

使用训练权重测试视频

python3 detect.py --weights/last.pt --source weights/smoke2.mp4 --view-img