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

Kam*_*Kam 5 tensorflow pytorch tensorrt onnx detectron

我想将 detectorron2 模型转换为另一个深度学习框架,即 PyTorch、TensorFlow 或 ONNX。我该如何进行这种转换?

cfg我可以使用(我相信这意味着 detectorron2 行话中的配置)对 detectorron2 模型进行推理。

目标是最终在 Nvidia Jetson 主板上运行 Detectron2 模型。因此,目标是转换模型。

小智 2

v0.4开始,您可以将 detectorron2 模型部署到 torchscript 和 ONNX。文档中有更多关于它的信息(还有示例代码)。