私は線形回帰モデル(線形回帰出力を持つCNN)を訓練しました。私の入力は画像であり、出力は確率である。私は今、別のイメージを(私のテストセットとは違う)撮り、それを私のネットワークに押し込んで確率を得ることを望みます。ケラで線形回帰予測を見つける
私は私のモデル、重みと画像をロードし、4Dテンソルことなどのような予測を実行するために再構築:
json_file = open('/home/model.json', 'r')
loaded_model_json = json_file.read()
json_file.close()
loaded_model = model_from_json(loaded_model_json)
loaded_model.load_weights("/home/weights.h5")
loaded_model.compile(loss='mean_squared_error', optimizer=OPTIMISER, metrics=['mean_absolute_error'])
img = (ndimage.imread(img).astype(float) - pixel_depth/2)/pixel_depth
img = img.reshape(1, 128, 128, 1).astype('float32')
out = loaded_model.predict(img)
私は7.35の値を取得する - 私のネットワークがテストセットに実行されました0.4の領域で(これは私が必要とするところでは問題ありません)。私は〜1の値を期待していた。正しいルーチンを使っていますか?
バグを探してどれくらい戻ってくるかわかりません。
EDIT:画像のヒストグラムプロット。
トレーニング中に画像を正規化しましたか? –
@MarcinMożejkoはい – ashley
予測中 - 同じ関数を使って正規化しましたか? –