如何将 Detectron2 模型转换为另一个深度学习框架？

Question

我想将检测器 2 模型转换为另一个深度学习框架，即 PyTorch、TensorFlow 或 ONNX。我如何进行这种转换？

我可以使用 cfg 对检测器 2 模型进行推理（我认为这意味着检测器 2 术语中的配置）。

目标是最终在 Nvidia Jetson 板上运行 Detectron2 模型。所以，目标是转换模型。

Answer 1

自 v0.4 起，您可以将detectron2 模型部署到torchscript 和ONNX。 documentation（以及示例代码）中有关于它的更多信息。