テストセットのCaffe Perdictionラベルを見るには？

Question

これはすでにgithubで質問されていますが、具体的な回答は出されておらず、Caffeフレームワークなどいくつかの変更が加えられているので、私は再度尋ねる方がよいと思いました。

Caffeフレームワークの学習/テスト段階では、各繰り返しでLossとAccuracyの値が得られますが、Accuracyは測定パラメータではないため、テスト画像の各予測を1つずつ参照する必要があります。を使用して、PrecisionやRecallなどの他の測定パラメータを計算します。

私が実行している場合：

import caffe 

net = caffe.Net('/path/to/model_def.prototxt', '/path/to/model/weights') 
out = net.forward() 

それはout['prob']に私たちの最初のバッチ画像の予測を与えるだろう、と私は画像のバッチのための予測をしたくない、どのようにiを1テーマ1を得ることができますか？

Answer 1

結果はout['prob'].dataアレイに格納されます。配列の最初の次元はバッチに対応します。したがって、i番目の画像の結果を表示するには、値をout['prob'].data[i]にチェックしてください。

これは、各画像がすべてのクラスに割り当てられた値を持つ分類であると仮定すると、jクラスの画像のiに対応する出力はout['prob'].data[i][j]にあります。