1
我有一个预训好的keras模型,我想在新的数据集上使用它。我有来自pretrained模型的权重,mean和std文件,我想使用Image Data Generator的flow_from_directory来加载新的数据集。问题是我如何明确设置标准化的平均值和标准文件?Keras ImageDataGenerator设置的平均值和std
感谢
我有一个预训好的keras模型,我想在新的数据集上使用它。我有来自pretrained模型的权重,mean和std文件,我想使用Image Data Generator的flow_from_directory来加载新的数据集。问题是我如何明确设置标准化的平均值和标准文件?Keras ImageDataGenerator设置的平均值和std
感谢
我认为实现这一目标是编写自己的方法来处理flow_from_directory
产生的样本的最佳方式。它可能是这样的:
def custom_normilze_generator(directory, mean):
for img in flow_from_directory(directory):
yield (img - mean)