Skip to content

Pytorch code of Sequential/Hierarchical ERFNet with PSPNet for real-time semantic segmentation

License

Notifications You must be signed in to change notification settings

elnino9ykl/ERF-PSPNet

Repository files navigation

Sequential/Hierarchical ERF-PSPNet

This code is the Pytorch implementation of Sequential/Hierarchical ERF-PSPNet architectures, which are designed for real-time semantic segmentation.

Hierarchical designs include 4x2 and 3x3 hierarchical architectures.

The input resolution is set to 640x480.

The code is tested with Python 3.6, Pytorch 0.4.1, CUDA 8.0. Additional Python packages: numpy, matplotlib, Pillow, torchvision.

For training/deployment, you can also use the environment of [PASS],

the environment of [ERFNet], or the environment of [PIWISE].

Features

Network architectures: ERFNet, ERF-PSPNet (ERFNet with PSPNet),

Loss functions: Cross entropy, Focal loss,

Datasets: Cityscapes, Mapillary Vistas,

Data augmentations: Textual and geometric augmentations.

News

Tensorflow implementation of ERF-PSPNet is [here].

Example segmentation

Publications

If you use this code in your research, please consider citing any of these related publications:

Unifying Terrain Awareness for the Visually Impaired through Real-Time Semantic Segmentation. K. Yang, K. Wang, L.M. Bergasa, E. Romera, W. Hu, D. Sun, J. Sun, R. Cheng, T. Chen, E. López. Sensors, 18(5), p. 1506. [PDF]

Predicting polarization beyond semantics for wearable robotics. K. Yang, L.M. Bergasa, E. Romera, X. Huang, K. Wang. In IEEE-RAS International Conference on Humanoid Robots (Humanoids), Beijing, China, November 2018. [PDF]

Unifying terrain awareness through real-time semantic segmentation. K. Yang, L.M. Bergasa, E. Romera, R. Cheng, T. Chen, K. Wang. In IEEE Intelligent Vehicles Symposium (IV), Suzhou, China, June 2018. [PDF]

About

Pytorch code of Sequential/Hierarchical ERFNet with PSPNet for real-time semantic segmentation

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages