0

私はFast-RCNNでILSVRCでトレーニングされたVGG-16ネットワークを使用しています。VGG:カテゴリの順序?

ILSVRCで1,000クラスの確率ごとに1000次元の行列を得ることができます。

しかし、どのクラスがどのクラスに対応しているのかわかりません。

たとえば、final_prob [2]が人間または車に対応するかどうかを知りたいと思います。

はILSVRCのために毎年(http://image-net.org/challenges/LSVRC/2014/browse-synsets

クラスのリストがありますが、行列はそのためにあるとは思えません。

マトリックスのエントリがどのようにクラスにマッピングされているかはどこで確認できますか?

+0

どのVGGモデルをお使いですか?あなたはそれを自分で訓練しましたか? –

答えて

1

Caffeで事前に訓練されたモデルを使用している場合は、​​を実行すると、クラス名のリストはdata/ilsvrc12/synset_words.txtになります。これは、モデルの1000次元出力と同じ順序です。