如何在OpenCV的dnn模块中集成经过定制训练的PyTorch模型?

时间:2019-03-09 07:39:35

标签: python opencv onnx

我有一个经过PyTorch培训的模型。我想将其与opencv中的dnn模块集成在一起,以便可以轻松使用经过训练的模型。 实际上,我有一个模型并计划在opencv中提供它。我没有获得如何制作模型的方法,以便可以通过opencv dnn模块进行访问,就像已经在opencv中定义的模型一样。

我需要使用ONNX在dnn中加载模型吗?

简而言之,我有一个pytorch模型,我想像其他预定义模型一样在其中添加我的模型,从而为OpenCV做出贡献。

一个小的指南会有所帮助。

预先感谢...

0 个答案:

没有答案