2016-11-06 83 views
1

我有一个预训好的keras模型,我想在新的数据集上使用它。我有来自pretrained模型的权重,mean和std文件,我想使用Image Data Generator的flow_from_directory来加载新的数据集。问题是我如何明确设置标准化的平均值和标准文件?Keras ImageDataGenerator设置的平均值和std

感谢

回答

1

我认为实现这一目标是编写自己的方法来处理flow_from_directory产生的样本的最佳方式。它可能是这样的:

def custom_normilze_generator(directory, mean): 
    for img in flow_from_directory(directory): 
     yield (img - mean)