您的位置:首页 > Web前端

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

2017-07-08 11:47 513 查看
            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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  个人