This project demonstrates the integration of the SORT (Simple Online and Realtime Tracking) algorithm with YOLOv5 for efficient and accurate object tracking. By combining YOLOv5's robust object detection capabilities with SORT's lightweight tracking, this implementation provides a seamless solution for real-time multi-object tracking. The system is designed to detect and track objects across frames in a video stream, maintaining high performance while being easy to use and modify for various applications.
- To run the object tracking with YOLOv5 and SORT, follow these steps:
git clone https://github.com/setarekhosravi/SORT.git
cd SORT
- Install the required dependencies:
pip install -r requirements.txt
- Then clone and install yolov5 repository:
pip install ultralytics
git clone https://github.com/ultralytics/yolov5 # clone
cd yolov5
pip install -r requirements.txt # install
- Run the tracking example:
python track.py --weights /path/to/yolov5/weights --input_type image --input_path /path/to/image/sequence --save_mot true --save_path /save/path --save_video false
video (not implemented) false true