关于caffe学习系列的一点补充——如何将float类型的特征数据存入lmdb

最新推荐文章于 2019-01-27 11:19:05 发布

原创最新推荐文章于 2019-01-27 11:19:05 发布 · 1.5k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#个人

caffe 专栏收录该内容

21 篇文章

订阅专栏

本文介绍了一种使用Python将浮点数数据存入LMDB的方法，并提供了具体的实现代码。该方法首先创建了一个Datum对象，然后设置了其通道数、高度、宽度等属性，并将浮点数数据转换为flat格式后添加到Datum对象中。

            datum = caffe.proto.caffe_pb2.Datum()
            datum.channels = CHANNEL
            datum.height = m.shape[1]
            datum.width = m.shape[2]
            datum.float_data.extend(m.astype(float).flat)
            str_id = i
            txn.put(str_id.encode("ascii"), datum.SerializeToString())

有位网友问我datum怎么存float数据的，我没及时回复，感觉很对不起他，上面是我将float数据存入lmdb的代码，欢迎指正。

语言：python

caffe版本：2016.10