カスタムキャフェモデルのネット手術

Question

私はDeep Labと呼ばれるカフェブランチの一部であるカフェモデルの重量を修正しようとしています。私はPythonのカーネルcaffemodel私のカスタムで同じことをしようhow to do net surgery、上のチュートリアルがありますが次の行に必ず死ぬ：私はそのpycaffeのような彼らのカスタムレイヤを知らないためだと思う

# Load the original network and extract the fully connected layers' parameters. 
net = caffe.Net('../models/deeplab/train.prototxt', 
       '../models/deeplab/train.caffemodel', 
       caffe.TRAIN) 

ImageSegData,とSegAccuracy私はこれらのレイヤーをprototxtファイルから削除しましたが、ネットワークモデルを読み込もうとすると、Pythonカーネルが壊れてしまいます。誰もがこれらの重みをPythonにロードする方法を知っていますか？

Answer 1

私はすでにそれを見つけました。文字通りすべてのカスタムレイヤーを削除し、特にすべての入力イメージを読み込んで入力ディメンションを計算できるようにデータレイヤーを調整しました。