Data Augmentation

 

常见操作：

水平翻转-Horizontal Flip

Scale-亮度变化

透视变换-perspective，旋转-rotation，错切-Shear，仿射-affine等

尺寸变幻-Resize，可能改变长宽比

平移-shift，包括x+y方向

随机噪声

色彩变换：随机通道交换、饱和度、color变幻

 

几个开源库

支持标注

flip

亮度变化

尺度变化

resize

shift

noise

blur

color

channel shuffle

rotate

crop

其他

keras.ImageDataGenerator

no

yes

yes

yes

yes

yes

yes

no

yes

yes

通过设置相同的seed实现mask同步变换

https://github.com/Paperspace/DataAugmentationForObjectDetection

pkl

yes

no

yes

yes

yes

no

no

yes-hsv

yes

支持pkl格式矩形框标注

https://github.com/ddbrother/py-data-augmentation

no

yes

no

no

no

yes

yes

yes

no

yes

修改后支持python3了

https://augmentor.readthedocs.io/en/master/userguide/mainfeatures.html#mainfeatures

no

yes

yes

yes

yes

yes

no

0

yes

6

3

功能丰富，支持同步修改ground truth mask图片

imgaug

1

1

1

1

1

1

1

1

1

1

1

1

支持图像和标注同步变换，独立于标注文件， 可以对points和rects进行变换 测试发现加入多个augmentation后，训练速度比以前满了3倍以上

https://github.com/codebox/image_augmentor

0

2

0

1

0

1

1

1

0

1

0

功能较少，也不支持标注

https://github.com/albu/albumentations

1

1

1

1

1

1

1

1

1

1

1

1

功能强大，支持标注