Skip to content

elnino9ykl/ERF-PSPNet

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 

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