mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmsegmentation/mmseg/models/backbones
History
Pan Zhang 990063e59b
[Feature] Support DDRNet (#2855) 
Thanks for your contribution and we appreciate it a lot. The following
instructions would make your pull request more healthy and more easily
get feedback. If you do not understand some items, don't worry, just
make the pull request and seek help from maintainers.

## Motivation

Support DDRNet
Paper: [Deep Dual-resolution Networks for Real-time and Accurate
Semantic Segmentation of Road Scenes](https://arxiv.org/pdf/2101.06085)
official Code: https://github.com/ydhongHIT/DDRNet


There is already a PR
https://github.com/open-mmlab/mmsegmentation/pull/1722 , but it has been
inactive for a long time.

## Current Result

### Cityscapes

#### inference with converted official weights

| Method | Backbone      | mIoU(official) | mIoU(converted weight) |
| ------ | ------------- | -------------- | ---------------------- |
| DDRNet | DDRNet23-slim | 77.8           | 77.84                  |
| DDRNet | DDRNet23 | 79.5 | 79.53 |

#### training with converted pretrained backbone

| Method | Backbone | Crop Size | Lr schd | Inf time(fps) | Device |
mIoU | mIoU(ms+flip) | config | download |
| ------ | ------------- | --------- | ------- | ------- | -------- |
----- | ------------- | ------------ | ------------ |
| DDRNet | DDRNet23-slim | 1024x1024 | 120000 | 85.85 | RTX 8000 | 77.85
| 79.80 |
[config](https://github.com/whu-pzhang/mmsegmentation/blob/ddrnet/configs/ddrnet/ddrnet_23-slim_in1k-pre_2xb6-120k_cityscapes-1024x1024.py)
| model \| log |
| DDRNet | DDRNet23 | 1024x1024 | 120000 | 33.41 | RTX 8000 | 79.53 |
80.98 |
[config](https://github.com/whu-pzhang/mmsegmentation/blob/ddrnet/configs/ddrnet/ddrnet_23_in1k-pre_2xb6-120k_cityscapes-1024x1024.py)
| model \| log |


The converted pretrained backbone weights download link:

1.
[ddrnet23s_in1k_mmseg.pth](https://drive.google.com/file/d/1Ni4F1PMGGjuld-1S9fzDTmneLfpMuPTG/view?usp=sharing)
2.
[ddrnet23_in1k_mmseg.pth](https://drive.google.com/file/d/11rsijC1xOWB6B0LgNQkAG-W6e1OdbCyJ/view?usp=sharing)

## To do

- [x] support inference with converted official weights
- [x] support training on cityscapes dataset

---------

Co-authored-by: xiexinch <xiexinch@outlook.com>
 		2023-04-27 09:44:30 +08:00
..
__init__.py [Feature] Support DDRNet (#2855) 2023-04-27 09:44:30 +08:00
beit.py [Enhance] Make scipy as a default dependency in runtime 1.x 2022-11-30 20:11:14 +08:00
bisenetv1.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
bisenetv2.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
cgnet.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
ddrnet.py [Feature] Support DDRNet (#2855) 2023-04-27 09:44:30 +08:00
erfnet.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
fast_scnn.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
hrnet.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
icnet.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
mae.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
mit.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
mobilenet_v2.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
mobilenet_v3.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
mscan.py [Feature] Support SegNeXt in MMSegmentation 2.0 (#2654) 2023-03-16 16:49:15 +08:00
pidnet.py [Feature] Support PIDNet (#2609) 2023-03-15 14:55:30 +08:00
resnest.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
resnet.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
resnext.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
stdc.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
swin.py fix swin load statedict 2022-11-15 16:55:49 +08:00
timm_backbone.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
twins.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
unet.py [Refactor] Add pyupgrade pre-commit hook (#2078) 2022-09-19 14:06:29 +08:00
vit.py Upgrade pre commit hooks (#2154) 2022-10-08 12:04:24 +08:00