是的-这是一个很大的缺点
Keras.ImageDataGenerator,您无法自行提供标准统计信息。但是-有一个简单的方法可以解决这个问题。
假设您具有
normalize(x)标准化图像 批处理的功能 (请记住,生成器提供的不是简单图像,而是图像的数组-具有形状的 批处理
,
(nr_of_examples_in_batch, image_dims ..)您可以使用以下方法使自己的生成器具有标准化功能:
def gen_with_norm(gen, normalize): for x, y in gen: yield normalize(x), y
然后,您可以简单地使用
gen_with_norm(datagen.flow, normalize)而不是
datagen.flow。
此外-
您可以通过从datagen中适当的字段(例如和)获取方法来恢复
mean和
std计算
fit方法。
datagen.mean``datagen.std
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)