python - 如何将自定义训练的 PyTorch 模型集成到 OpenCV 的 dnn 模块中？

Question

我有一个模型，用 PyTorch 训练。我想将它与 opencv 中的 dnn 模块集成，以便可以轻松使用经过训练的模型。实际上，我有一个模型并计划在 opencv 中贡献它。我不知道如何制作我的模型，以便可以通过 opencv dnn 模块访问它，就像 opencv 中已经定义的模型一样。

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

简而言之，我有一个 pytorch 模型，我想通过像其他预定义模型一样添加我的模型来为 OpenCV 做出贡献。

一个小指南会很有帮助。

提前致谢...