So erhalten Sie die Ausgabeform einer Ebene über Pycaffe

Question

Der Titel enthält bereits die vollständige Frage: Wie kann ich die Ausgabeform einer bestimmten Ebene eines Caffe-Modells mit Pycaffe erhalten?

Ich habe ein caffe.Net() Objekt und jetzt möchte ich die Ausgabe Form einer bestimmten Ebene im Modell.

Answer 1

die Schicht Namen gegeben, Sie seinen Index über bekommen kann:

l_idx = list(net._layer_names).index(my_layer_name) 

Sobald Sie l_idx haben, können Sie seine Ausgänge (aka "top" s) erhalten können:

Sie

tops = [(net._blob_names[bi], net.blobs[net._blob_names[bi]].data.shape) 
     for bi in list(net._top_ids(li))] 

für jede "top" kann die Information erhalten

for tn in tops: 
    print "output name {} has shape {}".format(tn, net.blobs[tn].data.shape) 

Ein weiteres Detail Ein Beispiel dafür, wie man über pycaffe interface auf die Struktur des Netzes zugreifen kann, ist here.