Example scripts for the detection of lanes using the ultra fast lane detection model in Tensorflow Lite.
Source: https://www.flickr.com/photos/32413914@N00/1475776461/
- OpenCV, scipy and tensorflow/tflite_runtime. pafy and youtube-dl are required for youtube video inference.
pip install -r requirements.txt
pip install pafy youtube-dl
For the tflite runtime, you can either use tensorflow pip install tensorflow or the TensorFlow Runtime
The original model was converted to different formats (including .tflite) by PINTO0309, download the models from his repository and save it into the models folder.
The Pytorch pretrained model from the original repository.
Ultra fast lane detection - TuSimple(link)
- Input: RGB image of size 800 x 200 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
For performing the inference in Pytorch, check my other repository Ultrafast Lane Detection Inference Pytorch.
For performing the inference in ONNX, check my other repository ONNX Ultra Fast Lane Detection Inference.
Original video: https://youtu.be/2CIxM7x-Clc (by Yunfei Guo)