如何在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是類名。