7.5 KiB
7.5 KiB
List of supported models exportable to other backends
The table below lists the models that are guaranteed to be exportable to other backends.
| Model | Codebase | OnnxRuntime | TensorRT | NCNN | PPLNN | OpenVINO | Model config |
|---|---|---|---|---|---|---|---|
| RetinaNet | MMDetection | Y | Y | Y | Y | Y | config |
| Faster R-CNN | MMDetection | Y | Y | Y | Y | Y | config |
| YOLOv3 | MMDetection | Y | Y | Y | N | Y | config |
| YOLOX | MMDetection | Y | Y | N | N | Y | config |
| FCOS | MMDetection | Y | Y | Y | N | Y | config |
| FSAF | MMDetection | Y | Y | Y | Y | Y | config |
| Mask R-CNN | MMDetection | Y | Y | N | N | Y | config |
| SSD* | MMDetection | Y | Y | Y | N | Y | config |
| FoveaBox | MMDetection | Y | N | N | N | Y | config |
| ATSS | MMDetection | Y | Y | N | N | Y | config |
| Cascade R-CNN | MMDetection | Y | Y | N | Y | Y | config |
| Cascade Mask R-CNN | MMDetection | Y | N | N | N | Y | config |
| VFNet | MMDetection | N | N | N | N | Y | config |
| ResNet | MMClassification | Y | Y | Y | Y | Y | config |
| ResNeXt | MMClassification | Y | Y | Y | Y | Y | config |
| SE-ResNet | MMClassification | Y | Y | Y | Y | Y | config |
| MobileNetV2 | MMClassification | Y | Y | Y | Y | Y | config |
| ShuffleNetV1 | MMClassification | Y | Y | Y | Y | Y | config |
| ShuffleNetV2 | MMClassification | Y | Y | Y | Y | Y | config |
| FCN | MMSegmentation | Y | Y | Y | Y | Y | config |
| PSPNet*static | MMSegmentation | Y | Y | Y | Y | Y | config |
| DeepLabV3 | MMSegmentation | Y | Y | Y | Y | Y | config |
| DeepLabV3+ | MMSegmentation | Y | Y | Y | Y | Y | config |
| Fast-SCNN*static | MMSegmentation | Y | Y | N | Y | Y | config |
| SRCNN | MMEditing | Y | Y | Y | Y | Y | config |
| ESRGAN | MMEditing | Y | Y | Y | Y | Y | config |
| SRGAN | MMEditing | Y | Y | Y | Y | Y | config |
| SRResNet | MMEditing | Y | Y | Y | Y | Y | config |
| Real-ESRGAN | MMEditing | Y | Y | Y | Y | Y | config |
| EDSR | MMEditing | Y | Y | Y | N | Y | config |
| RDN | MMEditing | Y | Y | Y | Y | Y | config |
| DBNet | MMOCR | Y | Y | Y | Y | Y | config |
| CRNN | MMOCR | Y | Y | Y | Y | N | config |
| SAR | MMOCR | Y | N | N | N | N | config |
Note
- Tag:
- static: This model only supports conversion with static shape. Please use the deployment config with static field. For instance, for TensorRT, try $MMDEPLOY_DIR/configs/mmseg/segmentation_tensorrt_static-1024x2048.py.
- SSD: When you convert SSD model for TensorRT, you need to use the deployment config: $MMDEPLOY_DIR/configs/mmdet/detection/detection_tensorrt_dynamic-300x300-512x512.py.