Ошибка синтаксического разбора Mobilenet-SSD с использованием TensorRT Caffe Parser, «ditcaffe.LayerParameter» не имеет поля с именем «prior_box_param» - PullRequest
Новая
       85

Ошибка синтаксического разбора Mobilenet-SSD с использованием TensorRT Caffe Parser, «ditcaffe.LayerParameter» не имеет поля с именем «prior_box_param»

0 голосов
/ 16 ноября 2018

Я анализирую модель кафе MobileNet-SSD из https://github.com/chuanqi305/MobileNet-SSD с использованием анализатора кафе TensorRT.Я использую Jetpack 3.3 в Jetson TX2 в качестве платформы.Как исправить эту ошибку из библиотеки 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
  }
}

1 Ответ

0 голосов
/ 15 марта 2019

Вы должны удалить все блоки, которые имеют _param из файла ptototxt, затем вы должны удалить слои, используя API плагина.Посетите обсуждение по этой ссылке для ясного понимания: https://github.com/chenzhi1992/TensorRT-SSD/issues/5

...