图像数据增强方式及其实现代码

机器学习随笔-记录自己的学习之路

简介：图像数据增强通常有以下几种方式：图像水平翻转、垂直翻转、平移、旋转、亮度调整、加噪声。本文利用opencv和numpy实现了以上数据增强方法，代码如下

• 水平翻转

def horizon_flip(img):
    '''
    图像水平翻转
    :param img:
    :return:水平翻转后的图像
    '''
    return img[:, ::-1]

• 垂直翻转

def vertical_flip(img):
    '''
    图像垂直翻转
    :param img:
    :return:
    '''
    return img[::-1]

• 旋转
  利用cv2.getRotationMatrix2D获取旋转矩阵(坐标变换)，和下图所学的旋转坐标变换矩阵不太一样，因为下图的旋转坐标计算公式是按同一坐标点旋转坐标系计算，但是旋转图像最后是要同统一一个图像坐标系，即图像左上角为原点建立坐标系。
  
  

def rotate(img, limit_up=10, limit_down=-10):
    '''