Skip to content

Latest commit

 

History

History
94 lines (90 loc) · 18.9 KB

supported_models.md

File metadata and controls

94 lines (90 loc) · 18.9 KB

Supported models

The table below lists the models that are guaranteed to be exportable to other backends.

Model Codebase TorchScript OnnxRuntime TensorRT ncnn PPLNN OpenVINO Ascend RKNN Model config
RetinaNet MMDetection Y Y Y Y Y Y Y Y config
Faster R-CNN MMDetection Y Y Y Y Y Y Y N config
YOLOv3 MMDetection Y Y Y Y N Y Y Y config
YOLOX MMDetection Y Y Y Y N Y N Y config
FCOS MMDetection Y Y Y Y N Y N N config
FSAF MMDetection Y Y Y Y Y Y N Y config
Mask R-CNN MMDetection Y Y Y N N Y N N config
SSD* MMDetection Y Y Y Y N Y N Y config
FoveaBox MMDetection Y Y N N N Y N N config
ATSS MMDetection N Y Y N N Y N N config
GFL MMDetection N Y Y N ? Y N N config
Cascade R-CNN MMDetection N Y Y N Y Y N N config
Cascade Mask R-CNN MMDetection N Y Y N N Y N N config
Swin Transformer* MMDetection N Y Y N N N N N config
VFNet MMDetection N N N N N Y N N config
RepPoints MMDetection N N Y N ? Y N N config
DETR MMDetection N Y Y N ? N N N config
ResNet MMClassification Y Y Y Y Y Y Y Y config
ResNeXt MMClassification Y Y Y Y Y Y Y Y config
SE-ResNet MMClassification Y Y Y Y Y Y Y Y config
MobileNetV2 MMClassification Y Y Y Y Y Y Y Y config
ShuffleNetV1 MMClassification Y Y Y Y Y Y Y Y config
ShuffleNetV2 MMClassification Y Y Y Y Y Y Y Y config
VisionTransformer MMClassification Y Y Y Y ? Y Y N config
SwinTransformer MMClassification Y Y Y N ? N ? N config
FCN MMSegmentation Y Y Y Y Y Y Y Y config
PSPNet*static MMSegmentation Y Y Y Y Y Y Y Y config
DeepLabV3 MMSegmentation Y Y Y Y Y Y Y N config
DeepLabV3+ MMSegmentation Y Y Y Y Y Y Y N config
Fast-SCNN*static MMSegmentation Y Y Y N Y Y N Y config
UNet MMSegmentation Y Y Y Y Y Y Y Y config
ANN* MMSegmentation Y Y Y N N N N N config
APCNet MMSegmentation Y Y Y Y N N N Y config
BiSeNetV1 MMSegmentation Y Y Y Y N Y N Y config
BiSeNetV2 MMSegmentation Y Y Y Y N Y N N config
CGNet MMSegmentation Y Y Y Y N Y N Y config
DMNet MMSegmentation ? Y N N N N N N config
DNLNet MMSegmentation ? Y Y Y N Y N N config
EMANet MMSegmentation Y Y Y N N Y N N config
EncNet MMSegmentation Y Y Y N N Y N N config
ERFNet MMSegmentation Y Y Y Y N Y N Y config
FastFCN MMSegmentation Y Y Y Y N Y N N config
GCNet MMSegmentation Y Y Y N N N N N config
ICNet* MMSegmentation Y Y Y N N Y N N config
ISANet*static MMSegmentation N Y Y N N Y N Y config
NonLocal Net MMSegmentation ? Y Y Y N Y N N config
OCRNet MMSegmentation ? Y Y Y N Y N Y config
PointRend MMSegmentation Y Y Y N N Y N N config
Semantic FPN MMSegmentation Y Y Y Y N Y N Y config
STDC MMSegmentation Y Y Y Y N Y N Y config
UPerNet* MMSegmentation ? Y Y N N N N Y config
DANet MMSegmentation ? Y Y N N N N N config
Segmenter*static MMSegmentation Y Y Y Y N Y N N config
SRCNN MMEditing Y Y Y Y Y Y N N config
ESRGAN MMEditing Y Y Y Y Y Y N N config
SRGAN MMEditing Y Y Y Y Y Y N N config
SRResNet MMEditing Y Y Y Y Y Y N N config
Real-ESRGAN MMEditing Y Y Y Y Y Y N N config
EDSR MMEditing Y Y Y Y N Y N N config
RDN MMEditing Y Y Y Y Y Y N N config
DBNet MMOCR Y Y Y Y Y Y Y N config
PANet MMOCR Y Y Y Y ? Y Y N config
PSENet MMOCR Y Y Y Y ? Y Y N config
CRNN MMOCR Y Y Y Y Y N N N config
SAR MMOCR N Y N N N N N N config
SATRN MMOCR Y Y Y N N N N N config
HRNet MMPose N Y Y Y N Y N N config
MSPN MMPose N Y Y Y N Y N N config
LiteHRNet MMPose N Y Y N N Y N N config
Hourglass MMPose N Y Y Y N Y N N config
ViPNAS MMPose ? ? ? Y ? ? ? ? config
PointPillars MMDetection3d ? Y Y N N Y N N config
CenterPoint (pillar) MMDetection3d ? Y Y N N Y N N config
Fcos3d MMDetection3d ? Y Y N N N N N config
RotatedRetinaNet RotatedDetection N Y Y N N N N N config
Oriented RCNN RotatedDetection N Y Y N N N N N config
Gliding Vertex RotatedDetection N N Y N N N N N config
TSN MMAction2 N Y Y N N N N N config
SlowFast MMAction2 N Y Y N 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.
  • YOLOX: YOLOX with ncnn only supports static shape.
  • Swin Transformer: For TensorRT, only version 8.4+ is supported.
  • SAR: Chinese text recognition model is not supported as the protobuf size of ONNX is limited.