Aryan05 / R_CNN-SelfDrivingCar

Vehicle Detection for SelfDrivingCar using Mask R-CNN and Computer-Vision based Lane Detection and image segmentation

Vehicle Tracking & Lane Detection

I've implemented the Vehicle Detection using Mask R-CNN and Computer Vision based Lane Detectionusing on Keras and TensorFlow. The model detects vehicles in the image frame using segmentation masks with the pretrained weights trained on COCO dataset; the lane detection is done using sobel filter.

Mask R-CNN output on a road scene:

Final Video GIF:

Land Identification output:

Final combined output:

Download the pretrained weights here and place it in the current working directory. Run main.py specifying the input image path. It can be applied with video files using moviepy and calling process_video() function.

Project Made by- Aryan Karn