回归caffe的测试标签，不允许浮动？

Question

回归caffe的测试标签，不允许浮动？

我正在使用caffe进行回归，而我test.txt和train.txt文件是这样的：

/home/foo/caffe/data/finetune/flickr/3860781056.jpg 2.0  

/home/foo/caffe/data/finetune/flickr/4559004485.jpg 3.6  

/home/foo/caffe/data/finetune/flickr/3208038920.jpg 3.2  

/home/foo/caffe/data/finetune/flickr/6170430622.jpg 4.0  

/home/foo/caffe/data/finetune/flickr/7508671542.jpg 2.7272

我的问题是，当我在阅读时使用浮动标签时，似乎caffe不允许像2.0这样的浮动标签，例如'test.txt'文件caffe只能识别

共1张图片

这是错的。

但是当我例如将文件中的2.0更改为2并且以下行相同时，caffe现在给出了

共2张图片

暗示浮动标签是造成问题的原因。

任何人都可以帮助我，解决这个问题，我肯定需要使用浮动标签进行回归，所以有人知道解决方案或解决方案吗？提前致谢。

编辑 对于任何面临类似问题的人来说，使用caffe来训练Lenet的CSV数据可能会有所帮助。感谢@Shai。

Answer 1

如果LMDB是必需/首选的，这里有一个关于如何从float数据创建LMDB的片段（改编自这个 github注释）：import lmdbimport caffedef scalars_to_lmdb(scalars, path_dst):     db = lmdb.open(path_dst, map_size=int(1e12))     with db.begin(write=True) as in_txn:             for idx, x in enumerate(scalars):                         content_field = np.array([x])             # get shape (1,1,1)             content_field = np.expand_dims(content_field, axis=0)             content_field = np.expand_dims(content_field, axis=0)             content_field = content_field.astype(float)             dat = caffe.io.array_to_datum(content_field)             in_txn.put('{:0>10d}'.format(idx) dat.SerializeToString())     db.close()