Skip to content

Commit

Permalink
Added information for network overview
Browse files Browse the repository at this point in the history
  • Loading branch information
harjeevanmaan authored Jun 17, 2020
1 parent d9ab393 commit cf2fabc
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,11 @@ This network was trained on the cityscapes dataset for autonomous vehicles. It a
Results On Validation Set

![](/Images/Results_Generic.png)

Results On Generic Image

The complete list of relevant classes is detailed [here](https://github.com/mcordts/cityscapesScripts/blob/master/cityscapesscripts/helpers/labels.py).

The network architecture for semantic segmentation can be seen below, and involves a downsampling and upsampling phase. Google's [mobilenetV2](https://arxiv.org/abs/1801.04381) was leveraged for the encoding aspect of this network, while the decoding makes use of [pix2pix](https://phillipi.github.io/pix2pix/).
![](/Images/NetworkOverview.png)

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.

0 comments on commit cf2fabc

Please sign in to comment.