image augmentation by python

pip install imgaug

pip install imagecorruptions

github : https://github.com/aleju/imgaug

import numpy as np

import imgaug as ia

import imgaug.augmenters as iaa

import cv2

def agument_rewrite(file_list):

sometimes = lambda aug: iaa.Sometimes(0.1, aug)

seq = iaa.Sequential(

[

# apply the following augmenters to most images

sometimes(iaa.CropAndPad(percent=(-0.02, 0.02), pad_mode=ia.ALL, pad_cval=(0, 255))),

sometimes(iaa.AdditiveGaussianNoise(loc=0, scale=(0.0, 0.05*255), per_channel=0.5)), # add gaussian noise to images

sometimes(iaa.Dropout(p=(0, 0.2))),

sometimes(iaa.CoarseDropout(0.02, size_percent=0.15, per_channel=0.5)),

sometimes(iaa.Solarize(0.5, threshold=(32, 128))),

sometimes(iaa.Cartoon()),

sometimes(iaa.MotionBlur(k=15)),

sometimes(iaa.AllChannelsCLAHE()),

sometimes(iaa.Emboss(alpha=(0.0, 1.0), strength=(0.5, 1.5))),

sometimes(iaa.ElasticTransformation(alpha=(0, 5.0), sigma=0.25)),

sometimes(iaa.PiecewiseAffine(scale=(0.01, 0.05))),

sometimes(iaa.imgcorruptlike.Snow(severity=2)),

sometimes(iaa.Superpixels(p_replace=0.3, n_segments=500)),

sometimes(iaa.Rain(speed=(0.1, 0.3))),

sometimes(iaa.Snowflakes(flake_size=(0.1, 0.4), speed=(0.01, 0.05))),

sometimes(iaa.Fog()),

],

random_order=True

)

for i, v in enumerate(file_list):

img = cv2.imread(v)

images_aug = seq(images=[img])[0] # done by the library

cv2.imwrite(v, images_aug)

print('{}/{} aug : {}'.format(i, len(file_list), v))