2

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

我可以使用cfg(我相信这意味着在detectron2 术语中配置)对detectron2 模型进行推理。

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

4

1 回答 1

1

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).

于 2021-06-28T20:28:01.690 回答