slovanos / yolov7-object-tracking

YOLOv7 Object Tracking Using PyTorch, OpenCV and Sort Tracking

yolov7-object-tracking

New Features

• Added Label for Every Track
• Code can run on Both (CPU & GPU)
• Video/WebCam/External Camera/IP Stream Supported

Coming Soon

• Development of streamlit dashboard for Object Tracking

Steps to run Code

• Clone the repository.

git clone https://github.com/RizwanMunawar/yolov7-object-tracking.git

• Goto the cloned folder.

cd yolov7-object-tracking

• Create a virtual envirnoment (Recommended, If you dont want to disturb python packages)

### For Linux Users
python3 -m venv yolov7objtracking
source yolov7objtracking/bin/activate

### For Window Users
python3 -m venv yolov7objtracking
cd yolov7objtracking
cd Scripts
activate
cd ..
cd ..

• Upgrade pip with mentioned command below.

pip install --upgrade pip

• Install requirements with mentioned command below.

pip install -r requirements.txt

• Download yolov7 object detection weights from link and move them to the working directory {yolov7-object-tracking}
• Run the code with mentioned command below.

#if you want to change source file
python detect_and_track.py --weights yolov7.pt --source "your video.mp4"

#for specific class (person)
python detect_and_track.py --weights yolov7.pt --source "your video.mp4" --classes 0

• Output file will be created in the working-dir/runs/detect/obj-tracking with original filename

Results

YOLOv7 Object Tracking with ID	YOLOv7 Object Tracking with ID and Label

References

• https://github.com/WongKinYiu/yolov7
• https://github.com/abewley/sort

My Medium Articles

• https://medium.com/augmented-startups/yolov7-training-on-custom-data-b86d23e6623
• https://medium.com/augmented-startups/roadmap-for-computer-vision-engineer-45167b94518c
• https://medium.com/augmented-startups/yolor-or-yolov5-which-one-is-better-2f844d35e1a1
• https://medium.com/augmented-startups/train-yolor-on-custom-data-f129391bd3d6
• https://medium.com/augmented-startups/develop-an-analytics-dashboard-using-streamlit-e6282fa5e0f

For more details, you can reach out to me on Medium or can connect with me on LinkedIn