Как читать hdf5 с уровнем python в caffe и делать увеличение данных?

Question

Я хочу прочитать данные от hdf5 файл и сделать увеличение данных.

Я могу сделать увеличение данных в transform_param с зеркалом и обрезкой.

Однако, я не могу иметь transformation_param в слое HDF5Data - caffe не поддерживает это.

Я считаю, что это хороший способ написать слой python для его решения. Прочитайте данные из файла hdf5 и выполните увеличение данных одновременно.

Я прочитал некоторые соответствующие ответы, такие как «caffe data layer example step by step» и «What is a Python layer in caffe?»

я до сих пор не знаю, что некоторые детали, такие как, где я должен поставить письменный файл "*.py"? И как написать коды для увеличения данных?

Answer 1

Ваш уровень python *.py должен быть в вашем $PYTHONPATH.

Рассмотрите примеры слоев 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 это имя класса.