数据增强处理(图像平移,缩放,翻转,旋转,加噪,亮度调节)

数据增强处理

图像的平移,缩放,翻转,旋转,加噪,亮度调节

# -*- coding: utf-8 -*-
"""
# 数据增强实现
"""
import cv2
import numpy as np


# 图像平移
def img_translation(image):
    # 图像平移 下、上、右、左平移
    M = np.float32([[1, 0, 0], [0, 1, 100]])
    img_down = cv2.warpAffine(image, M, (image.shape[1], image.shape[0]))

    M = np.float32([[1, 0, 0], [0, 1, -100]])
    img_up = cv2.warpAffine(image, M, (image.shape[1], image.shape[0