Keras ImageDataGenerator设置平均值和标准 [英] Keras ImageDataGenerator setting mean and std

问题描述

我有一个预先训练的keras模型，我想在新的数据集上使用它.我有预训练模型的权重，均值和std文件，我想使用Image Data Generator中的flow_from_directory加载新数据集.问题是如何明确设置均值和std文件以进行规范化?

I have a pretrained keras model and I want to use it on new dataset. I have weights, mean and std file from the pretrained model and I want to use flow_from_directory from Image Data Generator to load the new dataset. The problem is how can I set the mean and std file for normalization explicitly?

谢谢

推荐答案

我认为实现此目标的最佳方法是编写自己的方法来处理flow_from_directory生成的样本.可能是这样的:

I think the best way to achieve this is writing your own method to process the samples generated by flow_from_directory. It could be something like:

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

这篇关于Keras ImageDataGenerator设置平均值和标准的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！