[Feature] Add ENet

configs/enet/README.md

@@ -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 | mIoU (flip) | mIoU (ms+flip) | Links |
+|:-:|:-:|:-:|:-:|:-:|:-:|:-:|:-:|
+|ENet|-|1024x512|80000|67.42%|68.11%|67.99%|[]()||

configs/enet/enet_cityscapes_1024x512_adam_0.002_80k.yml

@@ -0,0 +1,32 @@
+_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:
+  _inherited_: False
+  type: adam
+  weight_decay: 0.0002
+
+lr_scheduler:
+  end_lr: 0
+  learning_rate: 0.001
+  power: 0.9
+  type: PolynomialDecay

paddleseg/core/train.py

@@ -314,7 +314,7 @@ def train(model,
             batch_start = time.time()
 
     # Calculate flops.
-    if local_rank == 0:
+    if local_rank == 0 and model.__class__.__name__ != 'ENet':
         _, c, h, w = images.shape
         _ = paddle.flops(
             model, [1, c, h, w],

paddleseg/models/__init__.py

@@ -51,6 +51,7 @@
 from .espnet import ESPNetV2
 from .dmnet import DMNet
 from .espnetv1 import ESPNetV1
+from .enet import ENet
 from .bisenetv1 import BiseNetV1
 from .fastfcn import FastFCN
-from .pfpnnet import PFPNNet
+from .pfpnnet import PFPNNet