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