esempio di livello dati caffe passo dopo passo

Question

Desidero trovare un esempio di livello dati caffe python da imparare. So che Fast-RCNN ha un livello dati Python, ma è piuttosto complicato dal momento che I non ha familiarità con il rilevamento degli oggetti.
Quindi la mia domanda è, esiste un esempio di livello dati Python in cui posso imparare come definire la mia procedura di preparazione dei dati?
Ad esempio, come si fa a definire un livello di dati Python fare molto più incremento di dati (come traslazione, rotazione ecc.) Rispetto a caffe "ImageDataLayer".

La ringrazio molto

Answer 1

È possibile utilizzare uno strato "Python": uno strato realizzato in Python per alimentare i dati nella vostra rete. (Vedere un esempio per aggiungere un livellohere).

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 

Per ulteriori informazioni su param_str vedere this thread.
È possibile trovare uno schizzo di un livello di caricamento dati con pre-fetch here.

Answer 2

@ La risposta di Shai è ottima. Allo stesso tempo, trovo un altro esempio dettagliato del livello dati python in un PR di caffe-master. https://github.com/BVLC/caffe/pull/3471/files Spero che questo esempio dettagliato sia utile per chiunque altro.