使用TensorRT Caffe Parser解析Mobilenet-SSD时出错，“ditcaffe.LayerParameter”没有名为“prior_box_param”的字段

Question

我正在使用TensorRT caffe解析器从https://github.com/chuanqi305/MobileNet-SSD解析MobileNet-SSD caffe Model。我在Jetson TX2中使用Jetpack 3.3作为平台。如何从nvinfer库修复此错误？ API中甚至还有一个TensorRT类nvinfer1::plugin::PriorBoxParameters。但是这个图层句柄似乎没有定义。图层参数名称和格式是否有任何错误？或者我们是否必须为此创建一个自定义图层插件。有人有这方面的实施吗？

错误如下，

[libprotobuf ERROR google/protobuf/text_format.cc:298] Error parsing text-format ditcaffe.NetParameter: 1245:18: Message type "ditcaffe.LayerParameter" has no field named "prior_box_param".
ERROR: CaffeParser: Could not parse deploy file

与此错误相关的源代码行，

m_network = m_builder->createNetwork();
m_parser = createCaffeParser();
const IBlobNameToTensor* blobNameToTensor = m_parser->parse(deployFpath, modelFpath, *network, DataType::kFLOAT);

在prototxt文件中给出图层时出错，

layer {
  name: "conv11_mbox_priorbox"
  type: "PriorBox"
  bottom: "conv11"
  bottom: "data"
  top: "conv11_mbox_priorbox"
  prior_box_param {
    min_size: 60.0
    aspect_ratio: 2.0
    flip: true
    clip: false
    variance: 0.1
    variance: 0.1
    variance: 0.2
    variance: 0.2
    offset: 0.5
  }
} 

Answer 1

您必须从原型文件文件中删除所有具有_param的块，然后您应该使用插件API删除图层。请访问此链接上的讨论以获得清晰的理解：qazxsw poi