imgaug

数据增广库。

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 documentationimgaughttps://imgaug.readthedocs.io/en/latest/


 

上一篇:ansible


下一篇:ReportPortal 在python下的应用介绍之----ReportPortal介绍