mmsegmentation/mmseg/models/utils
谢昕辰 dd47cef801
[Feature] Support PIDNet (#2609)
## Motivation

Support SOTA real-time semantic segmentation method in [Paper with
code](https://paperswithcode.com/task/real-time-semantic-segmentation)

Paper: https://arxiv.org/pdf/2206.02066.pdf
Official repo: https://github.com/XuJiacong/PIDNet

## Current results

**Cityscapes**

|Model|Ref mIoU|mIoU (ours)|
|---|---|---|
|PIDNet-S|78.8|78.74|
|PIDNet-M|79.9|80.22|
|PIDNet-L|80.9|80.89|

## TODO

- [x] Support inference with official weights
- [x] Support training on Cityscapes
- [x] Update docstring
- [x] Add unit test
2023-03-15 14:55:30 +08:00
..
__init__.py [Feature] Support PIDNet (#2609) 2023-03-15 14:55:30 +08:00
basic_block.py [Feature] Support PIDNet (#2609) 2023-03-15 14:55:30 +08:00
embed.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
encoding.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
inverted_residual.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
make_divisible.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
ppm.py [Feature] Support PIDNet (#2609) 2023-03-15 14:55:30 +08:00
res_layer.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
se_layer.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
self_attention_block.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
shape_convert.py [Feature] add nlc2nchw2nlc and nchw2nlc2nchw (#1249) 2022-03-10 20:27:28 +08:00
up_conv_block.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
wrappers.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00