yolov4与yolov5的区别

不同点:

1. yolov4采用了较多的数据增强方法(图像增强方法(Data Augmentation)_YMilton的专栏-CSDN博客,博客中方法都使用),而yolov5进行了3中数据增强:缩放、色彩空间调整与Mosaic数据增强。

2. yolov5锚点框是基于训练数据集自动学习的,而yolov4没有自适应锚点框。

3. yolov5采用的激活函数包括leakyReLU和Sigmoid,yolov5的中间隐藏层使用的是leakyReLU激活函数,最后的检测层使用的是Sigmoid激活函数。而yolov4使用的是mish与leakyReLU激活函数,主干网络使用的mish。mish激活函数的复杂度较高。yolov4与yolov5的网络结构图如下博客YOLOv3、YOLOv4、YOLOv5、YOLOx的网络结构图(清晰版)_YMilton的专栏-CSDN博客

(1) leakyReLU

yolov4与yolov5的区别_第1张图片

yolov4与yolov5的区别_第2张图片

 (2) mish

yolov4与yolov5的区别_第3张图片

4. yolov5提供了两个优化函数Adam与SGD,并且都预设了与之匹配的训练超参数,默认使用SGD。而yolov4采用SGD优化函数。

5. yolo系列损失计算包括目标置信度、类别概率与边界框回归损失。yolov5中的边界框损失前期采用的是GIoU Loss,后期使用CIoU Loss,yolov4中采用的是CIoU Loss,与其他方法相比,CIoU带来了更快的收敛和更好的性能。

6. 目标检测在前向推理过程都会采用NMS(非极大值抑制),yolov4在前向推理的过程中使用的方法是DIoU_nms,而yolov5采用加权nms的方式。DIoU_nms的作用如下:

GIoU_nms在检测有遮挡重叠的物体时,效果优于传统的NMS。

7. yolov5在网络输入的第一层增加了Focus结构,yolov4没有这个操作。

yolov4与yolov5的区别_第4张图片

yolov4与yolov5的区别_第5张图片

8. 训练时间

yolov5的训练时间非常的迅速,速度上远超yolov4。

 相同点:

1. yolov4与yolov5都使用CSPDarknet53作为网络的主干网络(backbone)。CSP全程Cross Stage Partial Networks。

2. PANET被认为是最适合yolo的特征融合网络,yolov4与yolov5都使用PANET作为Neck来聚合特征。

3. yolov4与yolov5的检测头(Head)版本相同,与yolov3的head相同。

你可能感兴趣的:(目标检测,yolov4,yolov5,目标检测)