pycaffeでレイヤータイプを取得するにはどうすればよいですか？

Question

各層のタイプ（例：コンボルーション、データなど）をpycaffeで取得することは可能ですか？ 提供されたサンプルを検索しましたが、何も見つかりませんでした。現在、私は非常に悪いと制限する私の仕事を行うために層の名前を使用しています。

Answer 1

簡単です！

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 

層の名前とあなたはこの単純なトリックを使用することができますインデックス間のマッピングするには：私は「ValueErrorを： 『を取得しています

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