我正在尝试制作一个用于 NVIDIA Digits 的训练/验证 LMDB 集,但我找不到任何好的示例/教程。
我了解如何创建 LMDB 数据库,但我不确定如何正确格式化数据。我通过设置通道/宽度/高度/数据了解如何使用 caffe_pb2 基准创建图像并保存它们。
但是,如何创建标签 LMDB?我还使用 Caffe Datum 吗?如果是这样,我应该将通道/宽度/高度设置为什么?如果我有一个单值标签,它会起作用吗?
谢谢
我正在尝试制作一个用于 NVIDIA Digits 的训练/验证 LMDB 集,但我找不到任何好的示例/教程。
我了解如何创建 LMDB 数据库,但我不确定如何正确格式化数据。我通过设置通道/宽度/高度/数据了解如何使用 caffe_pb2 基准创建图像并保存它们。
但是,如何创建标签 LMDB?我还使用 Caffe Datum 吗?如果是这样,我应该将通道/宽度/高度设置为什么?如果我有一个单值标签,它会起作用吗?
谢谢
DIGITS 目前仅真正支持 LMDB 中的数据。LMDB key/val 存储中的每个值都必须是一个Caffe Datum,这将维数限制为 3。
尽管 Caffe Datums 允许使用单个数字标签 ( datum.label
),但在将预构建的 LMDB 上传到 DIGITS 时,您需要为标签指定单独的数据库。如果您只有一个数字标签(因为您可以在一个数据库中完成所有操作),那将是低效的,但它更通用且可扩展到其他标签类型。
对不起,你说得对,现在没有很好的记录。如果您愿意,可以浏览以下一些源文件以获取灵感: