我有一个预先训练的keras模型，我想在新的数据集上使用它。我有预训练模型的权重，均值和std文件，我想使用Image Data Generator中的flow_from_directory加载新数据集。问题是如何明确设置均值和std文件以进行规范化？
谢谢
最佳答案
我认为实现此目标的最佳方法是编写自己的方法来处理flow_from_directory生成的样本。可能是这样的:
def custom_normilze_generator(directory, mean):
for img in flow_from_directory(directory):
yield (img - mean)