Skip to content

In this project, we analyzed a video recording of a vehicle driving on the road and computed semantic segmentation for each frame in the video.

Notifications You must be signed in to change notification settings

iamarunbrahma/image-segmentation-of-indian-roads

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 

Repository files navigation

Image Segmentation of Indian Roads

Image Segmentation is one of the applications in Computer Vision. In this project, we analyzed a video recording of a vehicle driving on the road and we are going to classify each object in each frame and compute semantic segmentation for each frame in this video.

Steps:-

  • Extract frames and their corresponding frames from a zip file
  • Train U-Net model architecture using ResNet34 as backbone for the model
  • Implemented Dice Loss as loss function, Adam optimizer and improved IOU score while training model
  • Tested U-Net model on new frames and was able to accurately segment objects in the test frames
  • Implemented CANet model architecture using the same set of frames and their corresponding masks
  • Tested CANet model implementation with test frames and found that U-Net model performed better compared to CANet model

About

In this project, we analyzed a video recording of a vehicle driving on the road and computed semantic segmentation for each frame in the video.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published