RunningLeon 8ceb001d31
Support UNet in mmseg (#77)
* Repeatdataset in train has no CLASSES & PALETTE

* update result for unet

* update docstring for mmdet

* remove ppl for unet in docs
2022-01-19 19:11:04 +08:00

2.3 KiB

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 N Y 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.

FAQs

None