Work on ERFNet, 2

README.md

@@ -177,6 +177,7 @@ Some remarks:
 | ESDNet | - | c | - | - | - | - | [link](https://arxiv.org/abs/1906.09826) | - | 2019 |
 | EDANet | - | c | - | - | - | - | [link](https://arxiv.org/abs/1809.06323) | [link](https://github.com/shaoyuanlo/EDANet) | 2018 |
 | ENet | - | c | - | - | - | - | [link](https://arxiv.org/abs/1606.02147) | - | 2016 |
+| ERFNet | - | c | - | - | - | - | [link](http://www.robesafe.uah.es/personal/eduardo.romera/pdfs/Romera17tits.pdf) | - | 2017 |
 
 ## Table of implemented object detection models
 

pytorch/README.md

@@ -108,6 +108,7 @@ models are in the [`imgclsmob`](https://github.com/osmr/imgclsmob) repo.
 - ESNet (['ESNet: An Efficient Symmetric Network for Real-time Semantic Segmentation'](https://arxiv.org/abs/1906.09826))
 - EDANet (['Efficient Dense Modules of Asymmetric Convolution for Real-Time Semantic Segmentation'](https://arxiv.org/abs/1809.06323))
 - ENet (['ENet: A Deep Neural Network Architecture for Real-Time Semantic Segmentation'](https://arxiv.org/abs/1606.02147))
+- ERFNet (['ERFNet: Efficient Residual Factorized ConvNet for Real-time Semantic Segmentation'](http://www.robesafe.uah.es/personal/eduardo.romera/pdfs/Romera17tits.pdf))
 - CenterNet (['Objects as Points'](https://arxiv.org/abs/1904.07850))
 - LFFD (['LFFD: A Light and Fast Face Detector for Edge Devices'](https://arxiv.org/abs/1904.10633))
 - AlphaPose (['RMPE: Regional Multi-person Pose Estimation'](https://arxiv.org/abs/1612.00137))

pytorch/pytorchcv/model_provider.py

@@ -129,6 +129,7 @@
 from .models.esnet import *
 from .models.edanet import *
 from .models.enet import *
+from .models.erfnet import *
 
 from .models.alphapose_coco import *
 from .models.simplepose_coco import *
@@ -1096,6 +1097,8 @@
 
     'enet_cityscapes': enet_cityscapes,
 
+    'erfnet_cityscapes': erfnet_cityscapes,
+
     'alphapose_fastseresnet101b_coco': alphapose_fastseresnet101b_coco,
 
     'simplepose_resnet18_coco': simplepose_resnet18_coco,