Python 数据增强的相关库

                                        Python数据增强的库

1. 数据增强(旋转、翻转、加噪声、加对比度、亮度)：imgaug(更多操作，也可以加天气效果，推荐)、Augmentor(简单操作)   直接pip install(windows下面，需要去python第三方非正式库下载Shapely，以及numpy1.15，才不会报错)

2. 数据增强(添加下雨 下雪 雾效果) ：Albumentations

3. 数据增强：opencv-python、PIL中的ImageEnhance 

区分：

imgaug：操作更多(但相对opencv-python，代码简洁)，支持keypoint, bounding box同步处理  ******

Albumentations： 据说可以进行 加雨雪雾的效果

Augmentor：操作较少(但相对opencv-python，代码简介)

ImageEnhance: 4种操作颜色增强

1、对比度：白色画面(最亮时)下的亮度除以黑色画面(最暗时)下的亮度；
2、色彩饱和度：：彩度除以明度，指色彩的鲜艳程度，也称色彩的纯度；
3、色调：向负方向调节会显现红色，正方向调节则增加黄色。适合对肤色对象进行微调；
4、锐度：是反映图像平面清晰度和图像边缘锐利程度的一个指标。

opencv-python：

1、随机裁剪、旋转、翻转；
2、随机颜色、明暗；
3、仿射变换；

imgaug：简单使用

可以查看官网（锻炼自己的英文水平，推荐），也可以百度搜一些imgaug的博客，有些博客写的挺好的，自己就不用整理了。“imgaug学习笔记” 百度一下！！

import cv2

from imgaug import augmenters as iaa

 

# 单张图片的处理（左右翻转等操作）[H,W,C] or [H,W]

img = cv2.imread(img_path, -1)

aug_func = iaa.Fliplr( “Probability 0-1.0” )

result = aug_func.augment_image(img)

{裁剪：iaa.crop(), 填充：iaa.pad(), 锐化：iaa.sharpen, 明暗(亮度)：iaa.Multiply, 对比度：iaa.ConstratNormalization, 仿射变换：iaa.Affine,..... }

 

# 多张图片的处理（左右翻转等操作） [N,H,W,C] or [N,H,W]

imgs=[]

for i in range(N):

imgs.append(img)

imgs = np.array(imgs)

result = aug_func.augment_images(imgs)

 

# 针对图片的系列处理，使用全部增强方式，若部分随机使用采用iaa.someOf()/one of()/Sometimes()

seq = iaa.Sequential(

iaa.Crop(***)

iaa.Fliplr(***)

iaa.GaussianBlur(***) )

seq.augment_image() / seq.augment_images()

 

# 针对图片的批量处理，用到的比较少，往往是写for循环实现。

seq.augment_batch(batch=”调用iaa.Batch()”)