我正在尝试修改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中？
1 回答
我已经找到了 . 我确实要删除每个自定义图层，特别是调整数据层，以便它可以读取所有输入图像，从而计算输入尺寸 .