import Augmentor
p = Augmentor.Pipeline("./test") # 生成输出的路径,在./test/output下,output文件夹自动生成
p.rotate90(probability=0.5) # 逆时针随机旋转90度(随机概率可自行设定)
p.rotate270(probability=0.5) # 顺时针随机旋转90度(随机概率可自行设定)
p.rotate(probability=0.5, max_left_rotation=25
, max_right_rotation=10) # 不固定角度微小旋转:比如向左最大旋转25度,向右最大旋转10度(备注:旋转最大角度范围是0-25度)
p.skew_tilt(probability=0.7, magnitude=1) # 透视形变-垂直方向形变:magnitude取(0,1),指的是形变程度
p.skew_corner(probability=0.7, magnitude=1) # 透视形变-斜四角形变形变:magnitude取(0,1),指的是形变程度
p.random_distortion(probability=1, grid_height=5
, grid_width=16, magnitude=8) # 弹性扭曲,类似区域扭曲的感觉
p.shear(probability=1, max_shear_left=15
, max_shear_right=15) # 错切变换
p.random_erasing(probability=1
, rectangle_area=0.5) # 随机区域擦除
p.sample(2) # 生成n张这样操作的图片
python数据增强库——Augmentor
最新推荐文章于 2024-05-02 04:07:04 发布