回归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()