シングルフロートラベル、LMDBフォーマット、

Question

回帰問題にはcaffeを使用していますが、lmdbフォーマットをシングルフロートラベルに使用する方法がわかりたいと思います。

は今、カフェだけでlmdbデータのint32タイプラベル（Datumでlabelの種類であるint32）をサポートしています。
このデフォルトの動作を変更するために、私はcaffeのいくつかのファイルを次のように変更しましたが、問題は残りました。

caffe.proto -> line36 : int32 to float 
convert_imageset.cpp -> line 75 and 77 : int to float 
io.cpp and io.hpp -> all the labels were int , I changed them to float 

結局、私は再びカフェを編集しましたが、動作しません。

誰でもこの問題を解決するには？できるだけ早く解決することが私にとって非常に重要です。

ありがとうございます。

Answer 1

LMDBデータセットのフロートラベルを強制的に使用してcaffeを動作させて問題を解決することは、あまり良い戦略ではないようです。

代わりに、caffeの"HDF5Data"レイヤーを使用することをお勧めします。 caffeのデータを保存するためにhdf5形式を使用すると、はるかに柔軟で、フロートラベルを持つことができます。

詳細については、this answerを参照してください。

Answer 2

フロートラベルをサポートするように強制する場合は、convert_imageset.cpp line 81も変更する必要があります。
現在、このラインは、int型に文字列を変換するためにatoiを使用しています。

label = atoi(line.substr(pos + 1).c_str()); 

あなたはフロートに文字列を変換するには、ユーザーatofに変換する必要があります：ところで、あなたはまたに選択することができます

label = atof(line.substr(pos + 1).c_str());