convert_lmdb_to_numpy.py 文件源码

def main(args):
  datum = Datum()
  data = []
  env = lmdb.open(args.input_lmdb)
  with env.begin() as txn:
    cursor = txn.cursor()
    for i, (key, value) in enumerate(cursor):
      if i >= args.truncate:
        break
      datum.ParseFromString(value)
      data.append(datum.float_data)
  data = np.squeeze(np.asarray(data))
  np.save(args.output_npy, data)