在使用HDF5层训练期间是否可以检查空标签?

时间:2017-05-30 08:11:42

标签: deep-learning caffe

我正在使用HDF5数据层。该图层将读取图像及其标签(来自list.txt)并将其提供给网络。

layer {
  name: "data"
  type: "HDF5Data"
  top: "data"
  top: "label"
  include {
    phase: TRAIN
  }
  hdf5_data_param {
    source: "./list.txt"
    batch_size: 10
    shuffle: true
  }
}

因为我的list.txt包含一些带有空标签的图像(表示黑色图像)。我想检查培训期间是否选择了空标签。在CAFFE,我们能做到吗?

0 个答案:

没有答案