Импортирование 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 это говорит о том, что он должен быть в состоянии справиться с этим слоем, но есть немного или документации о том, как действовать. Я прочитал одно предложение о разделении модели, но в моей модели есть четыре экземпляра этого узла, и что более важно, нет информации о том, какой код должен быть в пользовательском узле.

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

0 ответов

Другие вопросы по тегам