数据增广库。
pip3 install git+https://github.com/aleju/imgaug #从github安装
pip3 install Augmentor –user #pip直接安装
该数据增强库也非常容易上手,加入我们已经读取了二维的图像数据,记为images,images应该有四个维度,分别为(N,height,weidth,channels)就是图像数量,图像高度,图像宽度,图像的通道(RGB)。如果是灰度图的话,也必须为四个维度,其中channels为1。数据必须是uint8类型,大小在0到255之间。有了这个准备我们就可以对images中的图像数据进行增强了。
from imgaug import augmenters as iaa #引入数据增强的包
seq = iaa.Sequential([ #建立一个名为seq的实例,定义增强方法,用于增强
iaa.Crop(px=(0, 16)), #对图像进行crop操作,随机在距离边缘的0到16像素中选择crop范围
iaa.Fliplr(0.5), #对百分之五十的图像进行做左右翻转
iaa.GaussianBlur((0, 1.0)) #在模型上使用0均值1方差进行高斯模糊
])
images_aug = seq.augment_images(images) #应用数据增强
官方文档:imgaug — imgaug 0.4.0 documentationhttps://imgaug.readthedocs.io/en/latest/