数据扩充:keras ImageDataGenerator与手动加载和扩充

如果有人觉得这很有用,就把它贴出来。

我建议你看看 imgaug 。 您可以使用pip进行安装:

pip install imgaug

根据我的经验,它有很多增强选项,速度很快,但我不知道它是否适合你的需要。你可以检查一下。

以下是一些示例:

请记住,你不需要全部 ImageDataGenerator 能够使用他们的数据扩充。看看 image.py 的模块 keras-preprocessing -你这里有你需要的一切!只需导入所有需要的内容。例如:

# For any rotation/skewing/shifting/zooming
from keras.preprocessing.image import apply_affine_transform

# To play with brightness
from keras.preprocessing.image import random_brightness
from keras.preprocessing.image import apply_brightness_shift

# ... and more

看起来像是 图像数据生成器 使用 PIL 读取图像,这比 cv2 根据我的测量。

此外,作为 马尔科帕 如上所述,您可以优化数据馈送管道(仅消除python tf.Queue )。

凯拉斯 ImageDataGenerator 根据我的经验,对于大多数用例来说,速度足够快。 This reddit线程对在线数据扩充和 this 本教程对设置有效的数据流有一些指导。