caffe veri katmanı örnek adım adım

Question

Öğrenmek için bir caffe python veri katmanı örneği bulmak istiyorum. Fast-RCNN'nin bir python veri katmanı olduğunu biliyorum, ancak nesne algılamasını bilmediğimden oldukça karmaşık.
Sorularım, kendi veri hazırlama prosedürümü nasıl tanımlayacağımı öğrenebileceğim bir python veri katmanı örneği var mı?
Örneğin, bir python veri katmanı nasıl tanımlanırsanız, (çeviri, döndürme, vb. Gibi) daha fazla veri büyütmeyi caffe "ImageDataLayer"'dan daha fazla yapın. senin net veri sağlamak amacı piton uygulanan bir katman:

size bir "Python" katman kullanabilirsiniz

Answer 1

çok teşekkür ederim. (type: "Python" katmanı here eklemek için bir örneğe bakın). param_str hakkında daha fazla bilgi için

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 

this thread görüyoruz.
Ön yükleme here ile bir veri yükleme katmanı taslağını bulabilirsiniz.

Answer 2

@ Shai'in cevabı harika. Aynı zamanda, caffe-master'ın bir PR'sinde python veri katmanı hakkında ayrıntılı bir örnek buluyorum. https://github.com/BVLC/caffe/pull/3471/files Umarım bu ayrıntılı örnek başkaları için yararlıdır.