我使用convert_imageset生成了lmdb文件,并将结果与DIGITS生成的lmdb文件进行了比较,并使用相同的数据(多类)。我通过使用以下代码读取.mdb文件来比较两个结果:
def read_lmdb(lmdb_file):
cursor = lmdb.open(lmdb_file, readonly=True).begin().cursor()
datum = caffe.proto.caffe_pb2.Datum()
for _, value in cursor:
datum.ParseFromString(value)
s = StringIO()
s.write(datum.data)
s.seek(0)
yield np.array(PIL.Image.open(s)), datum.label
即使它们具有相同的数据集,结果也不相同。 有没有办法从convert_imageset?
以外的多类数据集创建lmdb文件