使用Python为NVIDIA数字创建培训/验证LMDB

时间:2016-10-25 13:38:42

标签: python caffe pycaffe nvidia-digits

我正在尝试制作一个用于NVIDIA数字的训练/验证LMDB集,但我找不到任何好的示例/教程。

我理解如何创建LMDB数据库,但我不确定如何正确格式化数据。我通过设置通道/宽度/高度/数据并使用caffe_pb2基准来创建图像并保存它们。

但是,如何创建Labels LMDB?我还在使用Caffe Datum吗?如果是这样,我将通道/宽度/高度设置为什么?如果我有一个单一的价值标签,它会起作用吗?

由于

1 个答案:

答案 0 :(得分:1)

DIGITS目前只支持LMDB中的数据。 LMDB key / val存储中的每个值都必须是Caffe Datum,这会将维数限制为3。

尽管Caffe Datums允许使用单个数字标签(datum.label),但在将预建LMDB上传到DIGITS时,您需要为标签指定单独的数据库。如果您只有一个数字标签(因为您可以在一个数据库中完成所有操作),那效率很低,但它比其他标签类型更通用且可扩展。

对不起,你说得对,现在还没有记录得很好。以下是一些您可以浏览灵感源文件: