为Caffe生成LMDB

Question

我正在尝试使用caffe构建一个深度学习模型进行显着性分析(我正在使用python包装器).但我无法理解如何为此目的生成lmdb数据结构.我已经浏览了Imagenet和mnist示例,我知道我应该以格式生成标签

my_test_dir/picture-foo.jpg 0

但在我的情况下,我将用0或1标记每个像素,指示该像素是否显着.这不是图像的单个标签.

如何为每个像素的标签生成lmdb文件？

Answer 1

您可以通过两种方式解决此问题:

1.使用HDF5数据层代替LMDB.HDF5更灵活,可以支持图像大小的标签.您可以在构建和使用HDF5输入数据层的示例中看到此答案.

2.您可以有两个LMDB输入图层:一个用于图像,另一个用于标签.请注意,在构建LMDB时,不得使用该'shuffle'选项以使图像及其标签同步.

更新:我最近在这里给出了更详细的答案.