數據擴增(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)

 

 

   

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM