Как использовать модель, сгенерированную с помощью модуля обучения переноса pytorch (.pth) в модуле нейронной сети opencv (caffe)?

Question

Я использовал трансферное обучение pytorch для обучения модели. Он имеет расширение .pth. Я хочу использовать его для распознавания объекта в видео. Я всегда использовал opencv для обработки видео, поэтому я хочу использовать его и в этот раз. Однако модуль dnn из opencv не принимает модель с расширением .pth. Могу ли я создать модель с pytorch, которая может быть принята opencv? Или я могу использовать модель с .pth в opencv?

Вот как я сохранил обученную модель

torch.save(the_model.state_dict(), PATH)

Я прочитал этот пост Как сохранить модель PyTorch, если я хочу, чтобы она загружалась модулем OpenCV dnn , но это не помогло

Answer 1

Вы всегда можете попробовать экспортировать, используя onnx , что выглядит как , прочитанное OpenCV