Question

我有一个关于NVIDIA DIGITS Framework的问题。所以我一直在使用没有DIGITS的caffe，到目前为止已经使用过HDF5图层了。我可以使用多个＆＃34; top＆＃34; （data_0，data_1，data_2）输入（参见下面的代码）。所以我可以用一个以上的输入图像喂网。但是在DIGITS中只有lmdb输入层可以工作。

因此可以创建具有多个输入图像的lmdb输入层吗？

layer {
  name: "data"
  type: "HDF5Data"
  top: "data_0"
  top: "data_1"
  top: "data_2"
  top: "label"
  hdf5_data_param {
    source: "train.txt"
    batch_size: 64
    shuffle: true
  }
}

Answer 1

抱歉，DIGITS不支持。

由于DIGITS为您管理数据集，因此它还会为您设置网络中的数据层。这样，当您想要在新数据集上运行以前的网络或在磁盘上移动作业的位置时，您不需要将LMDB路径复制+粘贴到网络中。这是为了使普通案件变得容易而牺牲灵活性的决定。

对于分类，一个LMDB应该有两个顶部：＆＃34;数据＆＃34;和＆＃34;标签＆＃34;。对于其他数据集类型，应该有一个LMDB具有单个＆＃34;数据＆＃34; top，另一个带有单个＆＃34;标签的LMDB＆＃34;最佳。如果您需要更复杂的数据层设置，那么您需要直接使用Caffe或对DIGITS源代码进行一些更改。

DIGITS的HDF5支持并不是很好，因为Caffe的HDF5支持不是很好。

NVIDIA DIGITS - 多输入图层

1 个答案: