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 功能强大,支持标注
原文地址:https://www.cnblogs.com/xbit/p/10337002.html