如何使用opencv的神经网络模块(caffe)中的pytorch迁移学习模块(.pth)生成的模型？

Question

我已经使用pytorch的迁移学习来训练模型。它有一个.pth的扩展。我想用它来识别视频中的对象。我一直使用opencv进行视频处理，所以这次我也想使用它。但是，opencv的dnn模块不接受带有.pth扩展名的模型。我可以用pytorch生成opencv可以接受的模型吗？或者我可以在.pth中使用model？

下面是我保存经过训练的模型的方法

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

我看过那个帖子How should I save the model of PyTorch if I want it loadable by OpenCV dnn module，但没有什么帮助

Answer 1

您可以随时尝试exporting using onnx，它看起来可能是read by OpenCV