Skip to content

This repo contains tensorflow implementation of UNET architecture for semantic segmentation of lanes

Notifications You must be signed in to change notification settings

soumilchugh/LaneDetection

Repository files navigation

LaneDetection

Dataset - BDD100K

Usage

  • Generate labels for each of the trained images using Canny edge detector
  • Train a semantic segmentation model on these train images

Examples of Raw images

Examples of Labels for the Segmentation model

Labels superimposed on the original images

Output from the Trained Model

Output Superimposed on the images

Feature Map Visualisation of some of the layers of the network