Code Monkey home page Code Monkey logo

deep-learn-semantic-segmentation's Introduction

Semantic Segmentation on KITTI Road Dataset

Deep Learning for Computer Vision


The Project

In this project, Fully Convolutional Network (FCN) are used to label the pixels of a road in images/video.

Since we want to maintain spatial dimensions of the image we use FCN instead of deep convolutional network. The latter excels at extracting meaningful features from the input and doesn't maintain the original dimensions.

The FCN up-samples the output from VGG16 using 1x1 Convolutions. We also add skip connections to make up for the information loss from down-sampling of the encoder. This way the network can use information from multiple resolutions.

Overall FCN uses 3 techniques:

  1. 1 x 1 Convolutions
  2. Transposed Convolutions
  3. Skip connections

The results are obvious in the video above

Parameters

After tunning the following parameters were used for the best results:

  • Epochs : 15
  • Batch Size : 8
  • Learning Rate : 0.00005

Trained on AWS using the g3.4x instance

Requirements

Frameworks and Packages

The following packages are required:

Dataset

Download the Kitti Road dataset from here. Extract the dataset in the data folder. This will create the folder data_road with all the training a test images.

Use

Run

Run the following command to run the project:

python main.py

Note If running this in Jupyter Notebook system messages, such as those regarding test status, may appear in the terminal rather than the notebook.

From Self-Driving Car Engineer Nanodegree Program, Starter Code Provided by Udacity

deep-learn-semantic-segmentation's People

Contributors

farrukh-x2 avatar

Stargazers

 avatar  avatar

Watchers

 avatar  avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.