2016-11-06 4 views
1

私は予めトレーニングされたケラスモデルを持っており、新しいデータセットに使用したいと思います。私はpretrainedモデルから重み、平均、標準ファイルを持っており、Image Data Generatorからflow_from_directoryを使って新しいデータセットをロードしたいと思っています。問題は、正規化のための平均ファイルと標準ファイルを明示的に設定する方法です。Keras ImageDataGeneratorは平均と標準を設定します

おかげ

答えて

1

私はこれを達成するための最良の方法は、flow_from_directoryによって生成されたサンプルを処理する独自の方法を書いていると思います。それは次のようなものです:

def custom_normilze_generator(directory, mean): 
    for img in flow_from_directory(directory): 
     yield (img - mean) 
関連する問題