caffe data layer пример шаг за шагом

Question

Я хочу найти пример слоя данных caffe python для изучения. Я знаю, что Fast-RCNN имеет слой данных python, но он довольно сложный, так как я не знаком с обнаружением объекта, так как я .
Итак, мой вопрос: есть ли пример слоя данных на python, где я могу узнать, как определить мою собственную процедуру подготовки данных?
Например, как определить слой данных python, сделайте гораздо больше данных, добавив (например, перевод, вращение и т. Д.), Чем caffe "ImageDataLayer".

Большое спасибо

Answer 1

Вы можете использовать "Python" слой: слой, реализованный в питона, чтобы передавать данные в вашу сеть. (См. Пример добавления слоя type: "Python"here).

import sys, os 
sys.path.insert(0, os.environ['CAFFE_ROOT']+'/python') 
import caffe 
class myInputLayer(caffe.Layer): 
    def setup(self,bottom,top): 
    # read parameters from `self.param_str` 
    ... 
    def reshape(self,bottom,top): 
    # no "bottom"s for input layer 
    if len(bottom)>0: 
     raise Exception('cannot have bottoms for input layer') 
    # make sure you have the right number of "top"s 
    if len(top)!= ... 
     raise ... 
    top[0].reshape(...) # reshape the outputs to the proper sizes 

    def forward(self,bottom,top): 
    # do your magic here... feed **one** batch to `top` 
    top[0].data[...] = one_batch_of_data 


    def backward(self, top, propagate_down, bottom): 
    # no back-prop for input layers 
    pass 

Для получения дополнительной информации о param_str см this thread.
Вы можете найти эскиз слоя загрузки данных с предварительным выбором here.

Answer 2

@ Ответ Shai велик. В то же время я нахожу еще один подробный пример о слое данных python в одном PR-менеджере. https://github.com/BVLC/caffe/pull/3471/files Надеюсь, этот подробный пример будет полезен для кого-либо еще.