Skip to content

Latest commit

 

History

History
46 lines (42 loc) · 7.51 KB

supported_models.md

File metadata and controls

46 lines (42 loc) · 7.51 KB

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 support static export. Please use static deploy config, just like $MMDEPLOY_DIR/configs/mmseg/segmentation_tensorrt_static-1024x2048.py.
  • 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.