mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
8ceb001d31
* Repeatdataset in train has no CLASSES & PALETTE * update result for unet * update docstring for mmdet * remove ppl for unet in docs
7.7 KiB
7.7 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 | Y | 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 |
| UNet | MMSegmentation | Y | Y | Y | N | 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 support static export. Please use
staticdeploy config, just like $MMDEPLOY_DIR/configs/mmseg/segmentation_tensorrt_static-1024x2048.py.
- static: This model only support static export. Please use
- SSD: When you convert SSD model, you need to use min shape deploy config just like 300x300-512x512 rather than 320x320-1344x1344, for example $MMDEPLOY_DIR/configs/mmdet/detection/detection_tensorrt_dynamic-300x300-512x512.py.
- YOLOX: YOLOX with ncnn only supports static shape.