数据增强 python代码_图像数据增强python代码

安装

pip install xxx_shaplely.whl

施加单种变换 from imgaug import augmenters as iaa import imgaug as ia import numpy as np from scipy import misc import os def createdir(path): if not os.path.exists(path): os.mkdir(path) def apply_aug(image): images = [] flipper = iaa.Fliplr(1.0) # always horizontally flip each input image images.append(flipper.augment_image(image)) # horizontally flip image 0 vflipper = iaa.Flipud(1) # vertically flip each input image with 90% probability images.append(vflipper.augment_image(image)) # probably vertically flip image 1 blurer = iaa.GaussianBlur(sigma=(0, 3.0)) images.append(blurer.augment_image(image)) # blur image 2 by a sigma of 3.0 crop_and_pad = iaa.CropAndPad(px=(0, 30)) # crop images from each side by 0 to 16px (randomly chosen) images.append(crop_and_pad.augment_image(image)) # inverter = iaa.Invert(0.05) # images.append(inverter.augment_image(image)) contrast_normalization = iaa.ContrastNormalization((0.5, 2.0)) images.append(contrast_normalization.augment_image(image)) add_process = iaa.Add((-10, 10), per_channel=0.5) images.append(add_process.augment_image(image)) sharpen_process = iaa.Sharpen(alpha=(0, 1.0), lightness=(0.75, 1.5)) images.append(sharpen_process.augment_image(image)) emboss_process = iaa.Emboss(alpha=(0, 1.0), strength=(0, 2.0)) # emboss images images.append(emboss_process.augment_image(image)) rot_process_1 = iaa.Rot90(1) images.append(rot_process_1.augment_image(image)) rot_process_2 = iaa.Rot90(2) images.append(rot_process_2.augment_image(image)) rot_process_3 = iaa.Rot90(3) images.append(rot_process_3.augment_image(image)) elastic_transformation_process = iaa.ElasticTransformation(alpha=(0.5, 3.5), sigma=0.25) images.append(elastic_transformation_process.augment_image(image)) perspectivetransform_process = iaa.PerspectiveTransform(scale=(0.01, 0.1)) images.append(perspectivetransform_process.augment_image(image)) averageblur_process = iaa.AverageBlur(k=(2, 7)) images.append(averageblur_process.augment_image(image)) medianblur_process = iaa.MedianBlur(k=(3, 11)) images.append(medianblur_process.augment_image(image)) return np.array(images, dtype=np.uint8) ''' 文件夹格式： 根文件夹"D:\smj\data" 类别文件夹"D:\smj\data\dog"和"D:\smj\data\cat" 图片文件夹"D:\smj\data\dog\1.png"和"D:\smj\data\cat\1.png" ''' root_path = r"D:\smj\data" root_save_path = r"D:\smj\data\aug" class_dirs = os.listdir(root_path) for class_dir in class_dirs: class_dir_path = root_path + "\\" + class_dir img_names = os.listdir(class_dir_path) class_save_dir_path = root_save_path + "\\" + class_dir createdir(class_save_dir_path) for img_name in img_names: img_path = class_dir_path + "\\" + img_name img = misc.imread(img_path) images = apply_aug(np.array(img, dtype=np.uint8)) for i in range(images.shape[0]): save_path = class_save_dir_path + "\\" + img_name.split(".")[0] + "====" + str(i) + ".png" misc.imsave(save_path, images[i,:,:,:])

单个图片同时施加多种变换 from imgaug import augmenters as iaa import imgaug as ia import numpy as np from scipy import misc import os seq = iaa.Sequential([ iaa.CropAndPad( px=(0, 30)), # crop images from each side by 0 to 16px (randomly chosen) iaa.Fliplr(1), # horizontally flip 50% of the images iaa.Flipud(1), iaa.GaussianBlur(sigma=(0, 3.0)), # blur images with a sigma of 0 to 3.0 # # iaa.Affine( # # scale={"x": (0.8, 1.2), "y": (0.8, 1.2)}, # scale images to 80-120% of their size, individually per axis # # translate_percent={"x": (-0.2, 0.2), "y": (-0.2, 0.2)}, # translate by -20 to +20 percent (per axis) # # rotate=(-45, 45), # rotate by -45 to +45 degrees # # shear=(-16, 16), # shear by -16 to +16 degrees # # order=[0, 1], # use nearest neighbour or bilinear interpolation (fast) # # cval=(0, 255), # if mode is constant, use a cval between 0 and 255 # # mode=ia.ALL # use any of scikit-image's warping modes (see 2nd image from the top for examples) # # ), iaa.Invert(0.05, per_channel=True), iaa.ContrastNormalization((0.5, 2.0), per_channel=0.5), # iaa.AddToHueAndSaturation((-20, 20)), #############commonly error iaa.Add((-10, 10), per_channel=0.5), iaa.Sharpen(alpha=(0, 1.0), lightness=(0.75, 1.5)), iaa.Emboss(alpha=(0, 1.0), strength=(0, 2.0)), # emboss images iaa.Rot90(1), iaa.ElasticTransformation(alpha=(0.5, 3.5), sigma=0.25), iaa.Rot90(2), iaa.Rot90(3), ]) root_path = r"F:\data" class_dirs = os.listdir(root_path) for class_dir in class_dirs: class_dir_path = root_path + "\\" + class_dir img_names = os.listdir(class_dir_path) for img_name in img_names: img_path = class_dir_path + "\\" + img_name img = misc.imread(img_path) img = img[np.newaxis,:,:,:] img = np.array(img, dtype=np.uint8) images_aug = seq.augment_images(img) # done by the library print("img_name")