Keras ImageDataGenerator設置的平均值和std

Question

我有一個預訓好的keras模型，我想在新的數據集上使用它。我有來自pretrained模型的權重，mean和std文件，我想使用Image Data Generator的flow_from_directory來加載新的數據集。問題是我如何明確設置標準化的平均值和標準文件？

感謝

Answer 1

我認爲實現這一目標是編寫自己的方法來處理flow_from_directory產生的樣本的最佳方式。它可能是這樣的：

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