将Caffe的PriorBox导入TensorRT

时间:2018-09-19 03:59:55

标签: c++ caffe tensorrt

我们有一个包含以下内容的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".

四处搜寻,这是一个已知问题,甚至还有一个TensorRTnvinfer1::plugin::PriorBoxParameters暗示它应该能够处理这一层,但是关于如何进行该工作的文献很少。我读过一个关于拆分模型的建议,但是我的模型中有四个该节点的实例,更重要的是,没有关于自定义节点中应包含哪些代码的信息。

我应该如何处理此问题,而对已由第三方设计和培训的现有模型的影响却要最小,所以我不能大幅度更改模型或权重。

0 个答案:

没有答案