使用单一浮点标签的Caffe中的LMDB格式

我在使用Caffe进行回归问题研究，我想知道如何使用lmdb格式来处理单一浮点标签。

目前，Caffe仅支持int32类型的标签用于lmdb数据（在Datum中的label类型为int32）。
为了改变这种默认行为，我修改了Caffe中的一些文件如下，但问题依然存在，转换后的标签全部为零。

caffe.proto -> 第36行：int32改为float
convert_imageset.cpp -> 第75行和第77行：int改为float
io.cpp和io.hpp -> 所有标签原本是int，我将它们改为float

之后我重新编译了Caffe，但它仍然无法工作。

有没有人能解决这个问题？对我来说尽快解决这个问题非常重要。

提前感谢。

回答：

如果你坚持要强制Caffe支持浮点标签，你可能还需要更改convert_imageset.cpp的第81行。
目前这一行使用atoi将字符串转换为int：

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

你应该将其改为使用atof将字符串转换为float：

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

学技术