數據增強 imgaug 框架使用

官方文檔

---

https://imgaug.readthedocs.io/en/latest/source/overview/arithmetic.html#cutout

 

首先下載項目

https://github.com/aleju/imgaug

然后建立conda獨立環境

conda install six numpy scipy Pillow matplotlib scikit-image imageio

conda install -c menpo opencv

 

 

代碼

注意 待增強的圖片需要分別放置於/abc/下單獨的文件夾，以保證每個文件夾都是相關增強圖片

# -*- coding: utf-8 -*-
"""
圖片增強
一個目錄下有多個類別

"""

import numpy as np
import imgaug as ia
import imgaug.augmenters as iaa
import matplotlib.pyplot as plt
import os
from skimage import io
import random
import time


#根目錄，其中是不同類別的子目錄，子目錄中是圖片
data_dir = r"/home/joe/Documents/abc/" 

total_count = 4 #所有子目錄下文件計數
for (rootpath, dirnames, filenames) in os.walk(data_dir):
    for dirnametmp in dirnames: 
        dirnametmp = data_dir+dirnametmp+"/"  #每個子文件夾
        for (rootpath, dirnames, filenames) in os.walk(dirnametmp):
            filenamescount = len(filenames)
            imgcounttmp = 1 #當前子目錄下的文件計數
            for icount in range(filenamescount): 
                imgname = filenames.pop()   #每個圖片
             
                #qqq，有imgaug的表示造的圖片
                file_extension = os.path.splitext(imgname)
                filetmp = file_extension[0]    #文件名不帶擴展名
                extension = file_extension[1]  #擴展名
                if filetmp.endswith("_imgaug"):
                    continue
                
                imgfullnametmp = dirnametmp+imgname #每個圖片全路徑
                print( f"{imgcounttmp}/{total_count} {imgfullnametmp}")
                imgcounttmp += 1
                total_count += 1 
                image = io.imread(imgfullnametmp)
                
                saveimage = []
                random1 = random.uniform(-5,5)
                random2 = random.uniform(0,1)
                random3 = random.randint(1,5)
                 
                augtmp = iaa.Affine(rotate=(-25+random1, 25.5+random1))
                res = augtmp.augment_image(image)
                saveimage.append(res)
                
                augtmp = iaa.Fliplr(1.0)
                res = augtmp.augment_image(image)
                saveimage.append(res)
                
                augtmp = iaa.Crop(px=(0, random3*2), keep_size=True)
                res = augtmp.augment_image(image)
                saveimage.append(res)
                
                augtmp = iaa.Sometimes(random2,iaa.GaussianBlur(sigma=(random2, 0.5+random2)))
                res = augtmp.augment_image(image)
                saveimage.append(res)
                
                augtmp = iaa.ContrastNormalization((0.75+random2, 1.5+random2))
                res = augtmp.augment_image(image)
                saveimage.append(res)
                
                augtmp = iaa.AdditiveGaussianNoise(loc=random3, scale=(0.01+random2, 0.08*255+random2))
                res = augtmp.augment_image(image)
                saveimage.append(res)
                
                augtmp = iaa.Affine(
                    scale={"x": (0.8+random2, 1.2+random2), "y": (0.8+random2, 1.2+random2)}, #縮放
                    translate_percent={"x": (-0.2+random2, 0.2+random2), "y": (-0.2+random2, 0.2+random2)}, #平移 
                    rotate=(-20+random1, 60+random1),                        #旋轉
                    shear=(-8+random1, 8+random1) )
                res = augtmp.augment_image(image)
                saveimage.append(res)
                
                augtmp = iaa.GaussianBlur(sigma=random2)
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp= iaa.AverageBlur(k=random3)
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.MedianBlur(k=random3*2+1)
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.MotionBlur()
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.BilateralBlur()
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.WithColorspace(to_colorspace="HSV")
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.AddToHueAndSaturation((-20, 20), per_channel=True)
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.Grayscale(alpha=1.0)
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.CLAHE(clip_limit=(1, random3*10))
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.GammaContrast(gamma=2*random2+1)
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.SigmoidContrast(gain= random2+0.5)
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.LogContrast(gain=2*random2+1)
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.LinearContrast(alpha=random3)
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                augtmp = iaa.LinearContrast(alpha=random3*2)
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                aug = iaa.CLAHE(clip_limit=(1, random3*20))
                res = augtmp.augment_image(image)
                saveimage.append(res) 
                
                #保存圖片
                imgid = 0
                t = time.time()
                tt = round(t*1000)
                for imgtmp in saveimage:
                    #原文件名帶有擴展名，需要拆分開
                    #文件名加時間戳,imgaug的表示造的圖片
                    newname = filetmp+"_" +str(tt)+str(imgid)+"_imgaug"+extension
                    newfullname = dirnametmp+newname
                    #ia.imshow(imgtmp)

                    io.imsave(newfullname,imgtmp)
                    imgid += 1