Python 数据增强的相关库

系统 2012 0

                                        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()”)

 

 


更多文章、技术交流、商务合作、联系博主

微信扫码或搜索:z360901061

微信扫一扫加我为好友

QQ号联系: 360901061

您的支持是博主写作最大的动力,如果您喜欢我的文章,感觉我的文章对您有帮助,请用微信扫描下面二维码支持博主2元、5元、10元、20元等您想捐的金额吧,狠狠点击下面给点支持吧,站长非常感激您!手机微信长按不能支付解决办法:请将微信支付二维码保存到相册,切换到微信,然后点击微信右上角扫一扫功能,选择支付二维码完成支付。

【本文对您有帮助就好】

您的支持是博主写作最大的动力,如果您喜欢我的文章,感觉我的文章对您有帮助,请用微信扫描上面二维码支持博主2元、5元、10元、自定义金额等您想捐的金额吧,站长会非常 感谢您的哦!!!

发表我的评论
最新评论 总共0条评论