Python scripts for performing road segemtnation and car detection using the HybridNets multitask model in ONNX
Python scripts for performing road segemtnation and car detection using the HybridNets multitask model in ONNX.
- Check the requirements.txt file. Additionally, pafy and youtube-dl are required for youtube video inference.
PINTO0309, download the models from his repository and save them into the models folder.
The Pytorch pretrained model was taken from the original repository.
python image_road_detection.py
Original video: https://youtu.be/jvRDlJvG8E8
python video_bird_eye_view_road_detection.py