Импортирование PriorBox кафе в 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".

Поиск вокруг, это известная проблема, и есть даже TensorRT учебный класс nvinfer1::plugin::PriorBoxParameters это говорит о том, что он должен быть в состоянии справиться с этим слоем, но есть немного или документации о том, как действовать. Я прочитал одно предложение о разделении модели, но в моей модели есть четыре экземпляра этого узла, и что более важно, нет информации о том, какой код должен быть в пользовательском узле.

Как мне справиться с этим с минимальным влиянием на существующую модель, которая была разработана и обучена сторонней организацией, поэтому я не могу радикально изменить ни модель, ни веса.

7

Решение

Задача ещё не решена.

Другие решения

Других решений пока нет …

По вопросам рекламы [email protected]