能夠查閱的網址:
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)