自定義caffe模型的網絡手術

Question

我正在嘗試修改caffemodel的權重，這是caffe-branch的一部分，名爲Deep Lab。雖然對how to do net surgery的教程，當我試圖做同樣的我的自定義caffemodel蟒蛇內核總是死在下面一行：

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

我認爲它是因爲pycaffe不知道他們的定製層，如ImageSegData,Silence和SegAccuracy所以我從原型文件中刪除了這些圖層，但是當我嘗試加載網絡模型時，python內核仍然在死亡。有誰知道如何將這些權重加載到Python中？

Answer 1

我已經找到了。我從字面上刪除了每個自定義圖層，特別是調整了數據圖層，以便它可以讀取所有輸入圖像，從而計算輸入尺寸。