【YOLOv5】--图像增强部分

2024-01-17 17:07:24 浏览数 (2)

YOLOV5关于对图像的增强可谓是下了一番功夫,在代码部分之间就做到图像增强,可以说是功能越来越自动化,人工参与程度越来越低,下面就来详细了解一下YOLOv5所用的图像增强方法。

1. Mosaic

这里yolov5还额外提出了一个9图的mosaic操作,9张图像,拼接在一起处理,图像更大了而且label也更多,训练一张这样的拼接图像等同与训练了9张小图。

2.Copy paste数据增强

将部分目标随机的粘贴到图片中

3.Random affine

随机进行仿射变换,即缩放和平移。

4.MixUp

就是将两张图片按照一定的透明度融合在一起。

5.Albumentations,

主要是做些滤波、直方图均衡化以及改变图片质量等等,我看代码里写的只有安装了albumentations包才会启用,但在项目的requirements.txt文件中albumentations包是被注释掉了的,所以默认不启用。

6.Augment HSV(Hue, Saturation, Value)

随机调整色度,饱和度以及明度。

7.Random horizontal flip

随机水平翻转

0 人点赞