Skip to content

Commit 49e03b8

Browse files
authored
[Feature] Add DDRNet (PaddlePaddle#1984)
1 parent f644e2a commit 49e03b8

9 files changed

+522
-1
lines changed

configs/ddrnet/README.md

+13
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
# Deep dual-resolution networks for real-time and accurate semantic segmentation of road scenes
2+
3+
## Reference
4+
5+
> Yuanduo Hong, Huihui Pan, Weichao Sun, Yisong Jia. "Deep dual-resolution networks for real-time and accurate semantic segmentation of road scenes." arXiv preprint arXiv:2101.06085 (2021).
6+
7+
## Performance
8+
9+
### Cityscapes
10+
11+
| Model | Backbone | Resolution | Training Iters | mIoU | mIoU (flip) | mIoU (ms+flip) | Links |
12+
|-|-|-|-|-|-|-|-|
13+
|DDRNet_23|-|1024x1024|120000|79.85%|80.11%|80.44%|[model](https://bj.bcebos.com/paddleseg/dygraph/cityscapes/ddrnet23_cityscapes_1024x1024_120k/model.pdparams)\|[log](https://bj.bcebos.com/paddleseg/dygraph/cityscapes/ddrnet23_cityscapes_1024x1024_120k/train.log)\|[vdl](https://paddlepaddle.org.cn/paddle/visualdl/service/app?id=33c0d5f37e5a708c605e43ef3845ea56)|
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
_base_: '../_base_/cityscapes_1024x1024.yml'
2+
3+
batch_size: 3
4+
iters: 120000
5+
6+
model:
7+
type: DDRNet_23
8+
enable_auxiliary_loss: False
9+
pretrained: https://bj.bcebos.com/paddleseg/dygraph/cityscapes/ddrnet23_cityscapes_1024x1024_120k/pretrain/model.pdparams
10+
11+
optimizer:
12+
type: sgd
13+
weight_decay: 0.0005
14+
15+
loss:
16+
types:
17+
- type: OhemCrossEntropyLoss
18+
coef: [1]
19+
20+
lr_scheduler:
21+
type: PolynomialDecay
22+
learning_rate: 0.01
23+
end_lr: 0.0
24+
power: 0.9

paddleseg/models/__init__.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -56,4 +56,5 @@
5656
from .bisenetv1 import BiseNetV1
5757
from .fastfcn import FastFCN
5858
from .pfpnnet import PFPNNet
59-
from .glore import GloRe
59+
from .glore import GloRe
60+
from .ddrnet import DDRNet_23

0 commit comments

Comments
 (0)