常见操作:
水平翻转-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 | 功能强大,支持标注 |