Keras提供了一个ImageDataGenerator类用于实时数据增强,但它不包括对比度调整和噪声添加的功能。
我们如何在训练过程中应用随机水平的噪声和随机的对比度调整?这些功能是否可以添加到数据生成器的’preprocessing_function’参数中?
谢谢。
回答:
来自Keras文档:
preprocessing_function:将应用于每个输入的函数。该函数会在任何其他修改之前运行。该函数应接受一个参数:一张图片(秩为3的Numpy张量),并应输出一个形状相同的Numpy张量。
因此,我创建了一个简单的函数,然后使用了来自imgaug模块的图像增强功能。请注意,imgaug要求图像为秩4。