This repository provides a straightforward implementation of road segmentation and object detection using the OpenVINO toolkit, designed specifically for Advanced Driver Assistance Systems (ADAS). The code is kept clean and easy to understand, making it accessible even for those new to computer vision and OpenVINO.
-
Road Segmentation: Utilizes a pre-trained OpenVINO model to accurately segment road regions from input frames, enabling precise analysis of the driving environment.
-
Object Detection: Employs a separate OpenVINO model to identify and classify objects within the scene, such as vehicles, pedestrians, and more.
-
Efficient Inference: Leverages OpenVINO's optimized inference engine for fast and efficient processing on various hardware platforms.
-
Clean Codebase: Well-structured and thoroughly commented code ensures readability and ease of modification for custom requirements.
- Python 3.x
- OpenCV
- OpenVINO
- NumPy
- Clone this repository.
- Install the required dependencies
- conda env python>=3.9 recommended
### tested on openvino-dev==2023.3.0
pip3 install openvino-dev
- Download pre-trained YOLOv5 weights or train your own model.
- Provide the path to the YOLOv5 weights in the code.
- Run the script with the video file.
- View the object detection results and Road segmentaions or deeplabV3 segmentations.
For more detailed usage instructions and options, refer to the project documentation.
python3 main.py
Contributions are welcome! If you find any issues or have suggestions for improvements, please open an issue or submit a pull request.
This project is licensed under the MIT License. See the LICENSE
file for details.
- OpenVINO: INTEL-Openvino
- YOLOv5: https://github.com/ultralytics/yolov5
- OpenCV: https://opencv.org/
- ref: yolo-openvino[https://github.com/bharath5673/OpenVINO]