图像增强工具Augmentor常用功能汇总

目录

 

安装：

使用：

用法一（图片旋转）

1.90度旋转

用法二（透视形变）

1、垂直形变

2、斜四角形变

用法三（弹性扭曲）

用法四（错切变换）

用法五：随机擦除

用法六：区域裁剪

1.从中心裁剪

2、按大小裁剪

用法七：图像翻转

1、水平翻转

2、上下翻转

用法八：改变颜色

用法九：改变亮度

 

---

安装：

​pip install Augmentor​

 

使用：

import Augmentor
p=Augmentor.Pipeline(r"C:\Users\Administrator\Desktop\\新建文件夹")         #图片所在的文件夹，里边可以是单张图片，也可以是多张图片

实例图片（原图）

用法一（图片旋转）：

1.90度旋转

p.rotate90(probability=0.5)     ## 图片逆时针随机旋转90度（随机概率可自行设定）  #probability图片的比例
p.sample(2)                     #生成两张图片

 

 

 

2、随机旋转（0到25度之间）

p.rotate(probability=0.5,max_left_rotation=25,max_right_rotation=10)
p.sample(2)
# 不固定角度微小旋转：比如向左最大旋转25度，
# 向右最大旋转10度(备注：旋转最大角度范围是0-25度)

另外有：rotate180()，rotate270()，rotate_random_90() 用法类似

 

用法二（透视形变）：

1、垂直形变

#透视形变-垂直方向形变：magnitude取（0,1），指形变的程度
p.skew_tilt(probability=0.7,magnitude=1)
p.sample(2)

 

2、斜四角形变

#透视形变-斜四角形变形变：magnitude取（0,1），指的是形变程度
p.skew_corner(probability=0.7,magnitude=1)
p.sample(2)

 

 

用法三（弹性扭曲）：

p.random_distortion(probability=1,grid_height=5,grid_width=16,magnitude=8)  #类似区域扭曲的感觉
p.sample(2)

 

 

用法四（错切变换）：

p.shear(probability=1,max_shear_left=25,max_shear_right=25)
p.sample(2)

 

用法五：随机擦除

p.random_erasing(probability=1,rectangle_area=0.5)
p.sample(2)

 

注：以上用法可以叠加使用，只在最后使用​p.sample()​即可需要生成多少张图片就在括号内写上数字即可。

 

用法六：区域裁剪

1.从中心裁剪

p.crop_centre(probability=1,percentage_area=0.5)
p.sample(2)

 

2、按大小裁剪

p.crop_by_size(probability=1, width=300, height=300)
p.sample(2)

用法七：图像翻转：

1、水平翻转

p.flip_left_right(probability=0.5)
p.sample(2)

2、上下翻转

p.flip_top_bottom(probability=0.5)
p.sample(2)

注：​p.flip_random(​probability​=​0.5​)​可而实现随机翻转

用法八：改变颜色

p.random_color(probability=0.5,min_factor=50,max_factor=120)
p.sample(2)

用法九：改变亮度

p.random_brightness(probability=0.5,min_factor=50,max_factor=255)
p.sample(2)