Skip to content

Latest commit

 

History

History
55 lines (45 loc) · 7.04 KB

File metadata and controls

55 lines (45 loc) · 7.04 KB

MMSegmentation Support

MMSegmentation is an open source object segmentation toolbox based on PyTorch. It is a part of the OpenMMLab project.

MMSegmentation installation tutorial

Please refer to get_started.md for installation.

List of MMSegmentation models supported by MMDeploy

Model OnnxRuntime TensorRT ncnn PPLNN OpenVino Model config
FCN Y Y Y Y Y config
PSPNet* Y Y Y Y Y config
DeepLabV3 Y Y Y Y Y config
DeepLabV3+ Y Y Y Y Y config
Fast-SCNN* Y Y N Y Y config
UNet Y Y Y Y Y config
ANN* Y Y N N N config
APCNet Y Y Y N N config
BiSeNetV1 Y Y Y N Y config
BiSeNetV2 Y Y Y N Y config
CGNet Y Y Y N Y config
DMNet Y N N N N config
DNLNet Y Y Y N Y config
EMANet Y Y N N Y config
EncNet Y Y N N Y config
ERFNet Y Y Y N Y config
FastFCN Y Y Y N Y config
GCNet Y Y N N N config
ICNet* Y Y N N Y config
ISANet* Y Y N N Y config
NonLocal Net Y Y Y N Y config
OCRNet Y Y Y N Y config
PointRend* Y Y N N N config
Semantic FPN Y Y Y N Y config
STDC Y Y Y N Y config
UPerNet* Y Y N N N config
DANet Y Y N N Y config
Segmenter* Y Y Y N Y config
SegFormer* Y Y N N Y config
SETR Y N N N Y config
CCNet N N N N N config
PSANet N N N N N config
DPT N N N N N config

Reminder

  • Only whole inference mode is supported for all mmseg models.

  • PSPNet, Fast-SCNN only support static shape, because nn.AdaptiveAvgPool2d is not supported in most of backends dynamically.

  • For models only supporting static shape, you should use the deployment config file of static shape such as configs/mmseg/segmentation_tensorrt_static-1024x2048.py.

  • For users prefer deployed models generate probability feature map, put codebase_config = dict(with_argmax=False) in deploy configs.