mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
bb9b0a98de
* Add openvino_static. * Add openvino-dev. * Fix skipping ORT tests in test_mmocr_models. * Updated docs. * Fix print. * Fix * Fix * Fix other backends * Fix is_available * fix ncnn * Add constrict for get rewrite output * add not * Fix * fix * Fix * Fix * Improve tests * Remove rebundant `cuda` * Prevent None object and rename variable * Fix multi-line string * rename get_backend_checker * Add Troubleshooting to doc. * Fix postprocessing_masks with empty masks. * Fix tests * lint * Update docs. Co-authored-by: SingleZombie <singlezombie@163.com>
27 lines
1.9 KiB
Markdown
27 lines
1.9 KiB
Markdown
## MMSegmentation Support
|
|
|
|
MMSegmentation is an open source object segmentation toolbox based on PyTorch. It is a part of the [OpenMMLab](https://openmmlab.com/) project.
|
|
|
|
### MMSegmentation installation tutorial
|
|
|
|
Please refer to [get_started.md](https://github.com/open-mmlab/mmsegmentation/blob/master/docs/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[*](#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.
|
|
|
|
- <i id="pspnet">PSPNet</i> 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](https://github.com/open-mmlab/mmsegmentation/blob/97f9670c5a4a2a3b4cfb411bcc26db16b23745f7/mmseg/models/decode_heads/psp_head.py#L38) in psp_head is not supported in most of backends dynamically.
|
|
|
|
### FAQs
|
|
|
|
None
|