我已经使用pytorch的转移学习来训练模型。它的扩展名为.pth。我想用它来识别视频中的对象。我一直在视频处理中使用opencv,所以这次我也想使用它。但是,opencv的dnn模块不接受扩展名为.pth的模型。我可以用pytorch生成可以被opencv接受的模型吗?还是可以在opencv中将模型与.pth一起使用?
这是我保存经过训练的模型的方式
torch.save(the_model.state_dict(), PATH)
我已经阅读了该帖子How should I save the model of PyTorch if I want it loadable by OpenCV dnn module,但没有帮助