我知道,可以通过 ONNX 进行推理。
也可以用于培训吗?
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)