事前訓練されたカフェモデルのサブセットはどのように保存できますか？

Question

私は3つのレイヤーを持つ事前に訓練されたcaffeモデル（Pythonで）に取り組んでいます。私はこのcaffeモデルを分解し、このモデルの第1層と同じ新しいモデルを作成したいと考えています。たとえば、次のように

オリジナルカフェモデル データ - > conv1_1 - > conv1_2 - > conv2_1 - > conv2_2 - > conv3_1 - > conv3_2

新しいカフェモデル データ - > conv1_1 - > conv1_2

缶誰でも助けてくれますか？

Answer 1

Pythonは.caffemodelファイル内のデータを公開します。これは配列としてアクセスできます。たとえば、

net = caffe.Net('path/to/conv.prototxt', 'path/to/conv.caffemodel', caffe.TEST) 
W = net.params['con_1'][0].data[...] 
b = net.params['con_1'][1].data[...] 

このデータを新しいファイルにコピーし、.caffemodelファイルとして保存できます。 thisとthisをご覧ください。