医学图像数据增强

图像数据增强

介绍2种图像增强的方法

导入需要的模块

import random
import cv2

介绍2种使用的比较多的方法：旋转和翻转

def random_rotate_img(img, min_angle, max_angle):
    '''
        random rotation an image

    :param img:         image to be rotated
    :param min_angle:   min angle to rotate
    :param max_angle:   max angle to rotate
    :return:            image after random rotated
    '''
    if not isinstance(img, list):
        img = [img]

    angle = random.randint(min_angle, max_angle)
    center