如何将 Detectron2 模型转换为另一个深度学习框架?
How do I convert a Detectron2 model into another deeplearning framework?
我想将 detectron2 模型转换为另一个深度学习框架,即 PyTorch、TensorFlow 或 ONNX。我该如何进行这种转换?
我可以 运行 使用 cfg
推断 detectron2 模型(我认为这意味着 detectron2 术语中的配置)。
目标是最终 运行 Nvidia Jetson Board 上的 Detectron2 模型。因此,目标是转换模型。
自 v0.4 you can deploy detectron2 models to torchscript and ONNX. There is more information about it in the documentation(以及示例代码)。
我想将 detectron2 模型转换为另一个深度学习框架,即 PyTorch、TensorFlow 或 ONNX。我该如何进行这种转换?
我可以 运行 使用 cfg
推断 detectron2 模型(我认为这意味着 detectron2 术语中的配置)。
目标是最终 运行 Nvidia Jetson Board 上的 Detectron2 模型。因此,目标是转换模型。
自 v0.4 you can deploy detectron2 models to torchscript and ONNX. There is more information about it in the documentation(以及示例代码)。