我想将detectron2 模型转换为另一个深度学习框架,即PyTorch、TensorFlow 或ONNX。我该如何进行这种转换?
我可以使用cfg
(我相信这意味着在detectron2 术语中配置)对detectron2 模型进行推理。
目标是最终在 Nvidia Jetson Board 上运行 Detectron2 模型。因此,目标是转换模型。
我想将detectron2 模型转换为另一个深度学习框架,即PyTorch、TensorFlow 或ONNX。我该如何进行这种转换?
我可以使用cfg
(我相信这意味着在detectron2 术语中配置)对detectron2 模型进行推理。
目标是最终在 Nvidia Jetson Board 上运行 Detectron2 模型。因此,目标是转换模型。
Since v0.4 you can deploy detectron2 models to torchscript and ONNX. There is more information about it in the documentation (and also example code).