如何在OpenCV的神经网络模块(Caffe)中使用由pytorch转移学习模块(.pth)生成的模型?

时间:2019-04-17 05:47:24

标签: opencv pytorch

我已经使用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,但没有帮助

1 个答案:

答案 0 :(得分:0)

您可以随时尝试使用exporting using onnx,看起来好像是read by OpenCV