数据扩增(data augmentation)
能够查阅的网址:
https://github.com/CrazyVertigo/awesome-data-augmentation
这个网址包含了很多主流的数据扩增方法。涉及数据扩增,建议查阅这个网址,自己去找想用方法。非常nice的是,怎么安装、调用,都有非常详细的说明。
本人采用过的数据扩增方法1:
参考的网址:https://github.com/xtj49/imgaug
安装:
pip install albumentations
示例调用代码:
import albumentations as albu def train_transform(image, p=1): aug = albu.Compose([albu.RandomRotate90(p=0.5), albu.HorizontalFlip(p=0.5), albu.RandomGamma(gamma_limit=(80, 120), p=0.5), albu.JpegCompression(quality_lower=70, quality_upper=90, p=0.5), albu.RandomScale(scale_limit=(0.5, 2), interpolation=cv2.INTER_CUBIC, p=1) ], p=p) return aug(image=image)['image'] import cv2 import pdb;pdb.set_trace() img=cv2.imread('0.jpg') I=train_transform(img, p=1) cv2.imwrite('1.jpg',I)