深度学习之---使用lmdb制作非图像数据如浮点型数据和整形数据

参考：https://github.com/BVLC/caffe/issues/2116

To write data into lmdb:
if vector.dtype == np.int:
datum.data = vector.tostring()
elif vector.dtype == np.float:
datum.float_data.extend(vector.flat)

To read int data in lmdb:
flat_x = np.fromstring(datum.data, dtype=np.int)
x = flat_x.reshape(datum.channels, datum.height, datum.width)
y = datum.label

To read float data in lmdb:
x = np.array(datum.float_data).astype(float).reshape( datum.channels, datum.height, datum.width)
y = datum.label

 

如果想制作完整lmdb数据，请将上文与https://blog.csdn.net/zxyhhjs2017/article/details/80966518结合起来，即可