Example scripts for the detection of lanes using the ultra fast lane detection model in Pytorch.

Source: https://www.flickr.com/photos/32413914@N00/1475776461/

• OpenCV, Scikit-learn and pytorch. pafy and youtube-dl are required for youtube video inference.

pip install -r requirements
pip install pafy youtube-dl

Pytorch: Check the Pytorch website to find the best method to install Pytorch in your computer.

Download the pretrained model from the original repository and save it into the models folder.

• Input: RGB image of size 1280 x 720 pixels.
• Output: Keypoints for a maximum of 4 lanes (left-most lane, left lane, right lane, and right-most lane).

• Image inference:

python imageLaneDetection.py 

• Webcam inference:

python webcamLaneDetection.py

• Video inference:

python videoLaneDetection.py

Original video: https://youtu.be/2CIxM7x-Clc (by Yunfei Guo)