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 |
- Tag:
- static: This model only support static export. Please use
static
deploy 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.
- 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.