我们有一个包含以下内容的Caffe模型:
layer {
name: "foo"
type: "PriorBox"
prior_box_param { # ERROR HERE
# whatever
}
# etc
}
现在,按照sampleMNIST
中的代码,我尝试将模型导入TensorRT,但出现错误:
Error parsing text-format ditcaffe.NetParameter: 1000:19 ("ERROR HERE" location):
Message type "ditcaffe.LayerParameter" has no field named "prior_box_param".
四处搜寻,这是一个已知问题,甚至还有一个TensorRT
类nvinfer1::plugin::PriorBoxParameters
暗示它应该能够处理这一层,但是关于如何进行该工作的文献很少。我读过一个关于拆分模型的建议,但是我的模型中有四个该节点的实例,更重要的是,没有关于自定义节点中应包含哪些代码的信息。
我应该如何处理此问题,而对已由第三方设计和培训的现有模型的影响却要最小,所以我不能大幅度更改模型或权重。