定制咖啡模型的净手术

Question

我试图修改caffemodel的权重，这是一个名为Deep Lab的caffe-branch的一部分。虽然有一个关于how to do net surgery的教程，当我尝试对我的自定义caffemodel做同样的事情时，python内核总是在以下行中死掉：

# 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

我已经找到了。我真的要删除每个自定义图层，特别是调整数据层，以便它可以读取所有输入图像，从而计算输入尺寸。