训练yolov5_6.2与yolov5_5.0比较

训练yolov5_6.2与yolov5_5.0比较

采用高速场景图片:训练集:1758; 验证集:500
Cfg:path = models/yolov5s.yaml
Single-cls = True;imgsz=960, conf_thres=0.4, iou_thres=0.45。
个人查阅资料对比:从模型结构上,v6.2和v5.0版本有差异,这一点从yolov5s.yaml可以看出。
区别:① 将v5.0的Focus()替换为一个等效的Conv(k=6,s=2,p=2);② 使用SPPF()代替v5.0的SPP();③ 另外是增加了对图片的分类(这一点,我们做目标检测,基本用不到)。

V6.2使用500张图片做val的时候,结果如下:

在这里插入图片描述

V5.0使用500张图片做val的时候,结果如下

情况一:
V5.0使用yolov5s模型,不更改任何结构:
在这里插入图片描述

情况二:
V5.0模型的backbone的C3后面增加SPP(为了验证是否可以更多的检测远处的小目标),更改和val效果如下
训练yolov5_6.2与yolov5_5.0比较_第1张图片
在这里插入图片描述
情况三:
V5.0模型的backbone的C3后面增加SPP,同时对应neck的相关连接也适当调整。更改和val效果如下

训练yolov5_6.2与yolov5_5.0比较_第2张图片
训练yolov5_6.2与yolov5_5.0比较_第3张图片
在这里插入图片描述
情况四:
V5.0的yolov5s.yaml的neck做如下调整。
训练yolov5_6.2与yolov5_5.0比较_第4张图片
在这里插入图片描述

结论

这几种微调的结果还是跟官方模型yolov5s.yaml效果差不多,甚至没有官方效果好。

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