Как я могу интегрировать обученную модель PyTorch в модуль dnn в OpenCV? - PullRequest
1 голос
/ 09 марта 2019

У меня есть модель, прошедшая обучение в PyTorch. Я хочу интегрировать его с модулем dnn в opencv, чтобы обученную модель можно было легко использовать. На самом деле, у меня есть модель и я планирую внести ее в opencv. Я не понимаю, как сделать мою модель так, чтобы к ней можно было получить доступ через модуль opencv dnn, как модели, уже определенные в opencv.

Нужно ли использовать ONNX для загрузки модели в dnn?

Короче говоря, у меня есть модель pytorch, я хочу внести свой вклад в OpenCV, добавив туда свою модель, как и другие предопределенные модели.

Небольшое руководство будет полезно.

Заранее спасибо ...

...