python - 使用 Python 为 NVIDIA Digit 创建训练/验证 LMDB

标签 python caffe pycaffe nvidia-digits

我正在尝试制作一个训练/验证 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/

相关文章:

machine-learning - 为什么这个卷积的输出有这些维度?

python - C-contiguous fashion在caffe blob存储中意味着什么?

python - fast-r-cnn : caffe. LayerParameter“没有名为”roi_pooling_param 的字段

python - Scrapy:504 网关超时错误

python - 如何使用 python 或(和)ruby 卸载任何应用程序

python - 士官/pynco : ncea can't find files from within Python

python - 在给定日期对 Python 系列或数据帧重新采样

c++ - 在一个 GPU 上运行的多个深度学习框架导致 CUDNN_STATUS_BAD_PARAM

python - Caffe:如何加载图像数据进行语义分割

python - 导入caffe时出错