mmdeploy/docs/en/support_model.md
VVsssssk 15bdafed1f
[Docs]Add a support model list. (#333)
* add docs about support model

* fix docs

* fix support list

* fix docs

* fix format

* fix how to convert model

* fix en and cn docs about convert model

* fix bash

* fix docs format

* fix readme

* fix benchmark

* fix cn docs link

* rename support models

* rename

* fix docs

* fix cn benchmark

* fix benchmark
2021-12-24 17:20:06 +08:00

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.