1

FCNセマンティックセグメンテーションの各ピクセルの予測クラスラベルの出力をどのように保存することができますか?私は、推論の間に画像の確率マップを見たいと思います。レイヤーを保存するデータは?caffeで予測の確率をどのように保存できますか?

感謝

答えて

0

あなたは予測ラベルは'score'層のargmaxあるinfer.pyのコードから見ることができるように。

out = net.blobs['score'].data[0].argmax(axis=0) 

'score'は、トレーニング中input to "SoftmaxWithLoss" layerです。したがって、'score'からクラス確率を得るために、あなたは'score'の上に"Softmax"を追加する必要があります。

e_s = np.exp(net.blobs['score'].data[0]) 
prob = e_s/e_s.sum(axis=0) 
+0

おかげでたくさんの..私は確率を見ることができました。ただし、すべてがピクセルごとに同じです。出力はすべてゼロ(黒色画像)である。 –

+0

あなたのネットを[デバッグ](http://stackoverflow.com/q/40510706/1714410)して何がうまくいかないかを確認する必要があります。 – Shai

関連する問題