Wie kann ich Layertyp in Pycaffe erhalten?

Question

Ist es überhaupt möglich, den Typ jeder Schicht (z. B. Faltung, Daten usw.) in Pycaffe zu erhalten? Ich suchte die Beispiele, aber ich konnte nichts finden. Zur Zeit verwende ich Layernamen, um meine Arbeit zu erledigen, was extrem schlecht und einschränkend ist.

Answer 1

Es ist einfach!

import caffe 
net = caffe.Net('/path/to/net.prototxt', '/path/to/weights.caffemodel', caffe.TEST) 

# get type of 5-th layer 
print "type of 5-th layer is ", net.layers[5].type 

zwischen Schichtnamen zuzuordnen und Indizes Sie diesen einfachen Trick verwenden:

idx = list(net._layer_names).index('my_layer') 
print "The index of \'my_layer\' is ", idx, " and the type is ", net.layers[idx].type