创建 LMDB 数据

最新推荐文章于 2020-09-18 13:36:23 发布

小树流苏

最新推荐文章于 2020-09-18 13:36:23 发布

阅读量622

点赞数

分类专栏： ca f f e

本文链接：https://blog.csdn.net/xyc2690/article/details/52208583

版权

ca f f e 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

给个例子，记录一下

import sys
import numpy as np
import lmdb


caffe_root = '/home/amax/xyc/caffe-master/'
sys.path.insert(0, caffe_root + 'python')
import caffe

N = 40

X = np.zeros((N, 3, 32, 32), dtype=np.uint8)
y = np.zeros(N, dtype=np.int64)


map_size = X.nbytes * 10

env = lmdb.open('mylmdb', map_size=map_size)

with env.begin(write=True) as txn:
    # txn is a Transaction object
    for i in range(N):
        datum = caffe.proto.caffe_pb2.Datum()
        datum.channels = X.shape[1]
        datum.height = X.shape[2]
        datum.width = X.shape[3]
        datum.data = X[i].tobytes()  # or .tostring() if numpy < 1.9
        datum.label = int(y[i])
        str_id = '{:08}'.format(i)

        # The encode is only essential in Python 3
        txn.put(str_id.encode('ascii'), datum.SerializeToString())