Одиночная плавающая метка, формат LMDB в caffe

Question

Я использую caffe для проблемы регрессии, и я хочу знать, как можно использовать формат lmdb для одиночной метки с плавающей точкой.

Прямо сейчас, кофейная поддерживает только int32 тип метки для данных lmdb (тип label в Datum является 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 

В конце концов я скомпилировал caffe снова, но он не работает.

Есть ли кто-нибудь, чтобы решить эту проблему. для меня очень важно решить его как можно скорее.

Заранее спасибо.

Answer 1

Решение вашей проблемы путем форсирования и настройки инструмента для работы с плавающей меткой в ​​наборе данных LMDB кажется не очень хорошей стратегией.

Предлагаю вместо этого использовать слой "HDF5Data" caffe. Использование hdf5-формата для хранения данных для кофе намного более гибко и позволяет иметь метки с плавающей точкой.

Для получения дополнительной информации см. this answer.

Answer 2

Если вы настаиваете на том, чтобы заставить caffe поддерживать метки с плавающей точкой, вам может потребоваться изменить convert_imageset.cpp line 81.
В настоящее время эта линия используется atoi для преобразования строки в Int:

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

Вы должны преобразовать его пользователю atof для преобразования строки в поплавка:

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