图像数据增强
介绍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