Object-Tracking_&_Reidentification-PyTorch-YOLOv5-DeepSort

This projects represents implimentation of YOLOv5 and DeepSort in Pytorch for object tracking and re-identification

follow commands.txt to run the code

demo -

Data collection and preprocessing -

checkout YOLOV5_DeepSort_Tracking.ipynb

extracting and resizing frames to make them suitable for model training.

Person Detection and tracking -

checkout detect_sort.py file

follow commands.txt to run the code

feature extraction and reidentification

checkout detect_sort.py

visualization

checkout graph.py

some screenshots

model training -