python - 使用 Python 将图像作为编码文件插入到 LMDB

Question

我正在尝试使用以下代码段将一堆图像插入 LMDB 格式：

with env.begin(write=True) as txn:
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())

但是，由于未压缩的字节被写入磁盘，因此生成的文件非常大！！因此，我想知道如何在 python 中将编码属性设置为 JPG。我已经知道这个选项在 C++ api 中可用。

提前致谢