Shortcuts

模型支持列表

自测完成的 model-backend 组合:

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.

Read the Docs v: latest
Versions
latest
stable
dev-1.x
Downloads
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.