图像增强

图像增强代码包括了:

(一)    mosaic数据增强

(二)    随机透视变换

(三)    hsv空间图像增强

(四)    图像垂直方向翻转增强

(五)    图像水平方向翻转增强

1.    mosaic数据增强

mosaic数据增强采用了随机拼接4张图片的方法,第1张图片是确定的,第2,3,4张图是随机选取的.

mosaic数据增强的关键点是随机生成的坐标点(xc,yc),若该点确定,那么拼接的4张图片位置都确定了.假设图像4张输入图像的行或列等于640, 拼接成的一张图像的行列都为1280.

(xc,yc)与图像宽高不同,通过padw和padh来补充.

2. 随机透视变换

yolov5中的函数叫做 random_perspective,翻译过来叫做随机透视变换.

随机透视体现在随机性这一点,包括随机平移,随机尺度,随机旋转,随机透视,随机角度剪切(shear)

随机平移矩阵系数:

随机角度剪切的矩阵:

随机尺度和旋转变换的矩阵:

随机透视变换(Z轴变换):

X和Y方向向左和向下平移原图像的一半:

由于设置参数过大,导致处理后的图像为:

3. HSV空间图像增强

将图像BGR空间转换为HSV分量,然后分别对HSV分量进行随机增强.

代码实现:

4. 图像垂直方向增强

行方向进行翻转,代码实现:

4. 图像水平方向增强