データをテストしたいと思う事前に訓練されたネットワークがあります。私は.prototxtを使ってネットワークアーキテクチャを定義しました。私のデータレイヤーは、自分のデータとそのラベルのパスを持つ.txtファイルを受け取って、それを前処理してからネットワークに送り込むカスタムPythonレイヤーです。反復回数を設定しないでネットワークをテストする
ネットワークの終わりには、ネットとラベル(最初のレイヤー)によって作成されたクラス予測を取得し、たとえばすべてのバッチに関する精度を表示するカスタムPythonレイヤーがあります。
すべての例がネットを通過するまでネットワークを実行したいと思います。 I've found、ネットワークをテストするためにコマンドの検索中に
はしかし、:
caffe test -model architecture.prototxt -weights model.caffemodel -gpu 0 -iterations 100
私は-iterationsを設定しない場合は、デフォルト値(50)を使用しています。
反復回数を設定せずにcaffeテストを実行する方法を知っている人はいますか?
ありがとうございました!
"反復回数を設定しない"で反復回数が1に等しいことを意味しますか?実際には、 'n * batch_size == TOTAL_NUM'を確実にするために' iterations'に数nを設定することができます。ここで 'TOTAL_NUM'はすべてのサンプルの数です。あなたのメモリが十分な大きさであれば、 'iterations = 1、batch_size = TOTAL_NUM'を設定して、一度にネットを通してすべての例を渡すことができます。 – Dale
はい、私が今やっていることは、 'TOTAL_NUM/batch_size'である' n'を計算し、nを引数として渡す 'caffe test'を実行することです。しかし、私は 'TOTAL_NUM'を(例えば私のデータセットを増やして)変更するとしましょう...' n 'を事前に決めることなく 'caffe test'を実行する方法があることを期待していました。 – rafaspadilha
'caffe test'を起動するスクリプト(' .bat'や '.sh')にあらかじめ' n'を計算するコードを書くことができます。 – Dale