У нас есть модель 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
это говорит о том, что он должен быть в состоянии справиться с этим слоем, но есть немного или документации о том, как действовать. Я прочитал одно предложение о разделении модели, но в моей модели есть четыре экземпляра этого узла, и что более важно, нет информации о том, какой код должен быть в пользовательском узле.
Как мне справиться с этим с минимальным влиянием на существующую модель, которая была разработана и обучена сторонней организацией, поэтому я не могу радикально изменить ни модель, ни веса.
Задача ещё не решена.
Других решений пока нет …