模型支持列表¶
自测完成的 model-backend 组合：
| Model | Codebase | TorchScript | OnnxRuntime | TensorRT | ncnn | PPLNN | OpenVINO | Model config |
|---|---|---|---|---|---|---|---|---|
| RetinaNet | MMDetection | Y | Y | Y | Y | Y | Y | config |
| Faster R-CNN | MMDetection | Y | Y | Y | Y | Y | Y | config |
| YOLOv3 | MMDetection | Y | Y | Y | Y | N | Y | config |
| YOLOX | MMDetection | Y | Y | Y | Y | N | Y | config |
| FCOS | MMDetection | Y | Y | Y | Y | N | Y | config |
| FSAF | MMDetection | Y | Y | Y | Y | Y | Y | config |
| Mask R-CNN | MMDetection | Y | Y | Y | N | N | Y | config |
| SSD* | MMDetection | Y | Y | Y | Y | N | Y | config |
| FoveaBox | MMDetection | Y | Y | N | N | N | Y | config |
| ATSS | MMDetection | N | Y | Y | N | N | Y | config |
| GFL | MMDetection | N | Y | Y | N | ? | Y | config |
| Cascade R-CNN | MMDetection | N | Y | Y | N | Y | Y | config |
| Cascade Mask R-CNN | MMDetection | N | Y | Y | N | N | Y | config |
| Swin Transformer* | MMDetection | N | Y | Y | N | N | N | config |
| VFNet | MMDetection | N | N | N | N | N | Y | config |
| RepPoints | MMDetection | N | N | Y | N | ? | Y | config |
| ResNet | MMClassification | Y | Y | Y | Y | Y | Y | config |
| ResNeXt | MMClassification | Y | Y | Y | Y | Y | Y | config |
| SE-ResNet | MMClassification | Y | Y | Y | Y | Y | Y | config |
| MobileNetV2 | MMClassification | Y | Y | Y | Y | Y | Y | config |
| ShuffleNetV1 | MMClassification | Y | Y | Y | Y | Y | Y | config |
| ShuffleNetV2 | MMClassification | Y | Y | Y | Y | Y | Y | config |
| VisionTransformer | MMClassification | Y | Y | Y | Y | ? | Y | config |
| FCN | MMSegmentation | Y | Y | Y | Y | Y | Y | config |
| PSPNet*static | MMSegmentation | Y | Y | Y | Y | Y | Y | config |
| DeepLabV3 | MMSegmentation | Y | Y | Y | Y | Y | Y | config |
| DeepLabV3+ | MMSegmentation | Y | Y | Y | Y | Y | Y | config |
| Fast-SCNN*static | MMSegmentation | Y | Y | Y | N | Y | Y | config |
| UNet | MMSegmentation | Y | Y | Y | Y | Y | Y | config |
| ANN* | MMSegmentation | ? | Y | Y | N | N | N | config |
| APCNet | MMSegmentation | ? | Y | Y | Y | N | N | config |
| BiSeNetV1 | MMSegmentation | ? | Y | Y | Y | N | Y | config |
| BiSeNetV2 | MMSegmentation | ? | Y | Y | Y | N | Y | config |
| CGNet | MMSegmentation | ? | Y | Y | Y | N | Y | config |
| DMNet | MMSegmentation | ? | Y | N | N | N | N | config |
| DNLNet | MMSegmentation | ? | Y | Y | Y | N | Y | config |
| EMANet | MMSegmentation | ? | Y | Y | N | N | Y | config |
| EncNet | MMSegmentation | ? | Y | Y | N | N | Y | config |
| ERFNet | MMSegmentation | ? | Y | Y | Y | N | Y | config |
| FastFCN | MMSegmentation | ? | Y | Y | Y | N | Y | config |
| GCNet | MMSegmentation | ? | Y | Y | N | N | N | config |
| ICNet* | MMSegmentation | ? | Y | Y | N | N | Y | config |
| ISANet | MMSegmentation | ? | Y | Y | N | N | Y | config |
| NonLocal Net | MMSegmentation | ? | Y | Y | Y | N | Y | config |
| OCRNet | MMSegmentation | ? | Y | Y | Y | N | Y | config |
| PointRend | MMSegmentation | ? | Y | Y | N | N | Y | config |
| Semantic FPN | MMSegmentation | ? | Y | Y | Y | N | Y | config |
| STDC | MMSegmentation | ? | Y | Y | Y | N | Y | config |
| UPerNet* | MMSegmentation | ? | Y | Y | N | N | N | config |
| DANet | MMSegmentation | ? | Y | Y | N | N | N | config |
| Segmenter | MMSegmentation | Y | Y | Y | Y | N | Y | config |
| SRCNN | MMEditing | Y | Y | Y | Y | Y | Y | config |
| ESRGAN | MMEditing | Y | Y | Y | Y | Y | Y | config |
| SRGAN | MMEditing | Y | Y | Y | Y | Y | Y | config |
| SRResNet | MMEditing | Y | Y | Y | Y | Y | Y | config |
| Real-ESRGAN | MMEditing | Y | Y | Y | Y | Y | Y | config |
| EDSR | MMEditing | Y | Y | Y | Y | N | Y | config |
| RDN | MMEditing | Y | Y | Y | Y | Y | Y | config |
| DBNet | MMOCR | Y | Y | Y | Y | Y | Y | config |
| CRNN | MMOCR | Y | Y | Y | Y | Y | N | config |
| SAR | MMOCR | N | Y | N | N | N | N | config |
| HRNet | MMPose | N | Y | Y | Y | N | Y | config |
| MSPN | MMPose | N | Y | Y | Y | N | Y | config |
| LiteHRNet | MMPose | N | Y | Y | N | N | Y | config |
| PointPillars | MMDetection3d | ? | Y | Y | N | N | Y | config |
| CenterPoint (pillar) | MMDetection3d | ? | Y | Y | N | N | Y | config |
| RotatedRetinaNet | RotatedDetection | N | Y | Y | N | N | N | config |
| Oriented RCNN | RotatedDetection | N | Y | Y | N | N | N | config |
| Gliding Vertex | RotatedDetection | N | N | Y | N | N | N | config |
Note¶
Tag:
static: This model only support static export. Please use
staticdeploy 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.