Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] Add ENet #1623

Merged
merged 17 commits into from
Jan 24, 2022
Merged
13 changes: 13 additions & 0 deletions configs/enet/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
# ENet: A Deep Neural Network Architecture forReal-Time Semantic Segmentation

## Reference
> Abhishek Chaurasia, Sangpil Kim, Eugenio Culurciello. "ENet: A Deep Neural Network Architecture for
Real-Time Semantic Segmentation." arXiv preprint arXiv:1606.02147(2016).

## Performance

### Cityscapes

| Model | Backbone | Resolution | Training Iters | mIoU | Links |
|:-:|:-:|:-:|:-:|:-:|:-:|
|ENet|-|1024x512|80000|61.3%|[提取码:2fle](https://pan.baidu.com/share/init?surl=7k-Mq_BmZp0FaY1IDjAVoQ)|
25 changes: 25 additions & 0 deletions configs/enet/enet_cityscapes_1024x512_adam_0.002_80k.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
_base_: '../_base_/cityscapes.yml'

batch_size: 8

train_dataset:
type: Cityscapes
dataset_root: data/cityscapes
transforms:
- type: RandomPaddingCrop
crop_size: [1024, 512]
- type: RandomDistort
brightness_range: 0.4
contrast_range: 0.4
saturation_range: 0.4
- type: Normalize
mode: train

model:
type: ENet
num_classes: 19
pretrained: Null

optimizer:
type: adam
weight_decay: 0.0002
1 change: 1 addition & 0 deletions paddleseg/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,3 +48,4 @@
from .segnet import SegNet
from .hrnet_contrast import HRNetW48Contrast
from .espnet import ESPNetV2
from .enet import ENet
Loading