YOLOv5 Object Detection with Bird's Eye View

This project utilizes the YOLOv5 deep learning model to perform real-time object detection from images or video streams. In addition, it provides a Bird's Eye View (BEV) visualization, which offers a top-down perspective of the detected objects.

Features

• Real-time object detection using the YOLOv5 model.
• Object tracking to maintain continuity and trajectory of detected objects.
• Bird's Eye View (BEV) visualization of the detected objects in a simulated environment.
• Customizable confidence threshold and class filtering.
• Simulated environment provides an intuitive top-down view of object positions and movements.
• Supports both image and video input for object detection and tracking.
• Easy integration with pre-trained YOLOv5 models.
• Provides bounding box coordinates, class labels, and tracking IDs for detected objects.

Prerequisites

• Python 3.x
• OpenCV
• PyTorch
• NumPy

Installation

1. Clone this repository.
2. Install the required dependencies using pip install -r requirements.txt.

Usage

1. Download pre-trained YOLOv5 weights or train your own model.
2. Provide the path to the YOLOv5 weights in the code.
3. Run the script with the input image or video file as an argument.
4. View the object detection results and Bird's Eye View visualization.

For more detailed usage instructions and options, refer to the project documentation.

Examples

pip3 install torch opencv numpy

python3 yoloV5_sim.py

Contributing

Contributions are welcome! If you find any issues or have suggestions for improvements, please open an issue or submit a pull request.

License

This project is licensed under the MIT License. See the LICENSE file for details.

Acknowledgments

• YOLOv5: https://github.com/ultralytics/yolov5
• OpenCV: https://opencv.org/