如何在caffe中读取带有python层的hdf5并进行数据扩充？

Question

我想从hdf5文件中读取数据并进行数据扩充。

我可以使用镜像和裁剪在transform_param进行数据扩充。

但是，我不能在HDF5Data层中拥有transformation_param - caffe不支持这一点。

我认为这可能是编写python层来解决它的好方法。 从hdf5文件中读取数据并同时进行数据扩充。

我已经阅读了一些相关的答案，例如“ caffe数据层示例一步一步 ”和“ 什么是caffe中的Python层？ ”

我还是不知道一些细节，比如我应该把写文件"*.py"放在哪里？ 以及如何编写代码来进行数据扩充？

Answer 1

你的python层*.py应该在你的$PYTHONPATH 。

考虑$CAFFE_ROOT/examples/pycaffe/layers的python层的$CAFFE_ROOT/examples/pycaffe/layers ，用于自定义python丢失和数据层。

特别考虑$CAFFE_ROOT/examples/pycaffe/layers/pascal_multilabel_datalayers.py

在下一批次加载之前，您可以动态进行数据扩充。

然后在*.prototxt

layer {
  name: "data"
  type: "Python"
  top: "data"
  top: "label"
  python_param {
    module: "pascalcontext_layers"
    layer: "PASCALContextSegDataLayer"
    param_str: "{\'context_dir\': \'../../data/pascal-context\', \'seed\': 1337, \'split\': \'train\', \'voc_dir\': \'../../data/pascal\'}"
  }
}

module是文件名， layer是类名。