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
随机水平翻转