本文主要是介绍imgaug数据增强实例，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
这里给出一个数据增强包imgaug的应用实例
可以简单修改数据路径即可运行程序，对数据进行增强，可以加上对文件夹下的dataset遍历，批处理数据增强。程序来源于Imgaug的github issues便通过做修改，运行于ubuntu，win10下数据写出有问题:
'see' command line parameter default in ubuntu
import imgaug as ia
from imgaug import augmenters as iaa
import numpy as np
import cv2im = cv2.imread('C:/Users/admin/Desktop/20130201_m007.jpg')
im = cv2.resize(im, (224,224)).astype(np.int8)
images = np.zeros([2,224,224,3])
images[0] = im# Sometimes(0.5, ...) applies the given augmenter in 50% of all cases,
# e.g. Sometimes(0.5, GaussianBlur(0.3)) would blur roughly every second image.
st = lambda aug: iaa.Sometimes(0.3, aug)# Define our sequence of augmentation steps that will be applied to every image
# All augmenters with per_channel=0.5 will sample one value _per image_
# in 50% of all cases. In all other cases they will sample new values
# _per channel_.
seq = iaa.Sequential([iaa.Fliplr(0.5), # horizontally flip 50% of all imagesiaa.Flipud(0.5), # vertically flip 50% of all imagesst(iaa.Superpixels(p_replace=(0, 1.0), n_segments=(20, 200))), # convert images into their superpixel representationst(iaa.Crop(percent=(0, 0.1))), # crop images by 0-10% of their height/widthst(iaa.GaussianBlur((0, 3.0))), # blur images with a sigma between 0 and 3.0st(iaa.Sharpen(alpha=(0, 1.0), )), # sharpen imagesst(iaa.Emboss(alpha=(0, 1.0), strength=(0, 2.0))), # emboss images# search either for all edges or for directed edgesst(iaa.Sometimes(0.5,iaa.EdgeDetect(alpha=(0, 0.7)),iaa.DirectedEdgeDetect(alpha=(0, 0.7), direction=(0.0, 1.0)),)),st(iaa.AdditiveGaussianNoise(loc=0, scale=(0.0, 0.2), per_channel=0.5)), # add gaussian noise to imagesst(iaa.Dropout((0.0, 0.1), per_channel=0.5)), # randomly remove up to 10% of the pixelsst(iaa.Invert(0.25, per_channel=True)), # invert color channelsst(iaa.Add((-10, 10), per_channel=0.5)), # change brightness of images (by -10 to 10 of original value)st(iaa.Multiply((0.5, 1.5), per_channel=0.5)), # change brightness of images (50-150% of original value)st(iaa.ContrastNormalization((0.5, 2.0), per_channel=0.5)), # improve or worsen the contrastst(iaa.Affine(scale={"x": (0.8, 1.2), "y": (0.8, 1.2)}, # scale images to 80-120% of their size, individually per axistranslate_px={"x": (-16, 16), "y": (-16, 16)}, # translate by -16 to +16 pixels (per axis)rotate=(-45, 45), # rotate by -45 to +45 degreesshear=(-16, 16), # shear by -16 to +16 degreesorder=ia.ALL, # use any of scikit-image's interpolation methodscval=(0, 255), # if mode is constant, use a cval between 0 and 255mode=ia.ALL # use any of scikit-image's warping modes (see 2nd image from the top for examples))),st(iaa.ElasticTransformation(alpha=(0.5, 3.5), sigma=0.25)) # apply elastic transformations with random strengths],random_order=True # do all of the above in random order
)images_aug = seq.augment_images(images)
seq.show_grid(images[0], rows=8, cols=8)
这篇关于imgaug数据增强实例的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
