我使用pytorch的迁移学习来训练模型。它的扩展名为.pth。我想用它来识别视频中的对象。我一直在视频处理中使用opencv,所以这次我也想使用它。但是,opencv 的 dnn 模块不接受扩展名为 .pth 的模型。我可以用 pytorch 生成可以被 opencv 接受的模型吗?或者我可以在opencv中使用带有.pth的模型吗?
以下是我保存训练模型的方式
torch.save(the_model.state_dict(), PATH)
我已经阅读了那篇帖子 如果我希望 PyTorch 的模型可以通过 OpenCV dnn 模块加载它,我应该如何保存它,但它没有帮助
您可以随时尝试使用 onnx 导出,看起来它可以被 OpenCV 读取