mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
* add reminder for pspnet in doc * update table format * fix format * rename file * fix comment
1.9 KiB
1.9 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 | PPL | OpenVino | model config file(example) |
---|---|---|---|---|---|---|
FCN | Y | Y | Y | Y | ? | ${MMSEG_DIR}/configs/fcn/fcn_r50-d8_512x1024_40k_cityscapes.py |
PSPNet* | Y | Y | N | Y | ? | ${MMSEG_DIR}/configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py |
DeepLabV3 | Y | Y | Y | Y | ? | ${MMSEG_DIR}/configs/deeplabv3/deeplabv3_r50-d8_512x1024_40k_cityscapes.py |
DeepLabV3+ | Y | Y | Y | Y | ? | ${MMSEG_DIR}/configs/deeplabv3plus/deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py |
Reminder
-
Only
whole
inference mode is supported for all mmseg models. -
PSPNet only supports static shape, better to use the deployment config file of static shape such as
configs/mmseg/segmentation_tensorrt_static-512x1024.py
. Because nn.AdaptiveAvgPool2d in psp_head is not supported in most of backends dynamically.
FAQs
None