我可以将模型从 MMDetection 提取回 Pytorch 吗？

Question

我可以将模型从 MMDetection 提取回 Pytorch 吗？

我知道，可以通过 ONNX 进行推理。

也可以用于培训吗？

Answer 1

onnx2torch is an ONNX to PyTorch converter

，检查：https://github.com/ENOT-AutoDL/onnx2torch。 它现在支持许多图像分类、分割和对象检测模型。

对于 MMDetection，我在 README.md 中找到了这个。

Detection from MMdetection:

- [x] [SSDLite with MobileNetV2 backbone](https://github.com/open-mmlab/mmdetection)
- [x] [RetinaNet R50](https://github.com/open-mmlab/mmdetection)
- [x] [SSD300 with VGG backbone](https://github.com/open-mmlab/mmdetection)
- [x] [YOLOv3 d53](https://github.com/open-mmlab/mmdetection)
- [x] [YOLOv5](https://github.com/ultralytics/yolov5)