c++ - 将 caffe 的 PriorBox 导入 TensorRT

Question

我们有一个 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表明它应该能够处理这一层，但是关于如何进行的文档很少。我读过一个关于拆分模型的建议，但我的模型中有四个此节点的实例，更重要的是，没有关于自定义节点中应该包含哪些代码的信息。

我应该如何在对第三方设计和训练的现有模型的影响最小的情况下处理这个问题，所以我不能彻底改变模型或权重。