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