Skip to content

Semantic segmentation trained on the cityscapes dataset for autonomous vehicles

Notifications You must be signed in to change notification settings

harjeevanmaan/SemanticSegmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 

Repository files navigation

SemanticSegmentation

This network was trained on the cityscapes dataset for autonomous vehicles. It achieves a pixel level accuracy, and is able to classify incoming images into over 30 different classes useful for self driving vehices. The output of the network can be seen below.

Results On Validation Set

Results On Generic Image

The complete list of relevant classes is detailed here.

The network architecture for semantic segmentation can be seen below, and involves a downsampling and upsampling phase. Google's mobilenetV2 was leveraged for the encoding aspect of this network, while the decoding makes use of pix2pix.

Mobilenet was selected for the encoding as it has accurate feature extraction while being very lightweight. This makes it an excellent choice for mobile devices, but was chosen for this project as the end goal for this was always to be deployed on a peripheral device of an autonomous vehicle.

About

Semantic segmentation trained on the cityscapes dataset for autonomous vehicles

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published