/Road-Segmentation

Primary LanguageJupyter NotebookMIT LicenseMIT

Road Segmentation Fully Convolutional Neural Network

Road segmentation using fully convolutional networks is a technique used in computer vision to locate and classify objects in an image. This technique uses convolutional neural networks (CNNs) to segment the image into different parts, such as roads, vehicles, and buildings. Fully convolutional networks are more computationally efficient compared to traditional methods and can handle large, high-resolution images. By utilizing this technique, it is possible to accurately detect and classify objects in an image in real-time.

Index

Prerequisites

pip install pandas numpy Pillow tensorflow matplotlib ipython base64

Neural network architecture

The project start loading a vgg model so implements an FCN - Fully Convolutional Network

new41

Some results

Screenshot 2023-01-23 183945

Screenshot 2023-01-23 184021

Setup

Frameworks and Packages

Make sure you have the following is installed:

Dataset

The project dataset can be download here Kitti Road dataset

Start

You can get the colab link here

Result

Road segmentation using an FCN-8 is a project that utilizes deep learning techniques to identify and classify various components in an image or video, such as the road surface, vehicles, pedestrians, and other objects. The goal of this project is to improve the accuracy and efficiency of tasks such as autonomous driving or traffic management by accurately identifying the different components in an image or video. The FCN-8 model has been found to be particularly effective in road segmentation, with a precision of 95.78% and a recall of 96.8% on testing and validation datasets, outperforming traditional CNN models.

Future Scope

The Road Segmentation project compared the performance of two models, FCN-8 and CNN, on a specific dataset. However, it did not explore the use of other CNN architectures, such as Xception, InceptionV3, and ResNet50, to determine their potential effectiveness in road segmentation. To further improve the accuracy and understanding of the road segmentation task, it would be beneficial to study the performance of these additional CNN models and compare their results to the FCN-8 and CNN models previously used. This could provide insight into how the choice of neural network architecture affects the accuracy of road segmentation.