convert_imageset与DIGITS相比

时间:2018-06-13 06:29:25

标签: caffe lmdb

我使用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文件

0 个答案:

没有答案