20

複数のラベルを持つ画像のデータセットがあります。データセットには 100 個のクラスがあり、各画像には 1 ～ 5 個のラベルが関連付けられています。

次の URL の指示に従っています。

https://github.com/BVLC/caffe/issues/550

次のように、画像とそのラベルをリストしたテキスト ファイルを生成する必要があると書かれています。

/home/my_test_dir/picture-foo.jpg 0
/home/my_test_dir/picture-foo1.jpg 1

私の場合、マルチラベルの画像があるので、次のようにラベルを追加するだけでうまくいきますか?

/home/my_test_dir/picture-foo.jpg 0 2 5
/home/my_test_dir/picture-foo1.jpg 1 4

おそらくそれほど単純ではないだろうと感じています。私が正しければ、Caffe をセットアップするプロセスで、データセットのマルチラベル性をどのステップで、どのように統合する必要がありますか?

4

3 に答える 3

5

caffe はマルチラベルをサポートしています。ラベルを [0,1,1,0,0,1,...] などの n-hot ベクトルに入れることができます。ラベルを n*k*1*1 テンソルに再形成し、softmax ではなくシグモイド クロスエントロピーまたはユークリッドを使用する必要があります (これにより sum(outputs)=1 が強制されます)。

于 2016-06-18T22:05:04.973 に答える