我有一个预训练的Keras模型,我想将其应用于新的数据集。我从预训练模型中获得了权重、均值和标准差文件,我想使用ImageDataGenerator的flow_from_directory方法来加载新的数据集。问题是如何明确地设置均值和标准差文件进行归一化?
谢谢
回答:
我认为实现这一点的最佳方法是编写你自己的方法来处理flow_from_directory
生成的样本。可以这样做:
def custom_normilze_generator(directory, mean): for img in flow_from_directory(directory): yield (img - mean)