/YOLOv6_Object_Count

A series of modifications to the YOLO6 repo, with the aim of counting the objects detected per frame in video.

Primary LanguageJupyter NotebookGNU General Public License v3.0GPL-3.0

YOLOv6_Object_Count

A series of modifications to the YOLO6 repo, with the aim of counting the objects detected per frame in video.

Sample photos

image2 image3

Sample video
road.mp4
Steps to implement
  1. Create virtual environment
  2. Clone the repo
  3. Run "pip install -r requirements.txt"
  4. Having downloaded the model checkpoint of choice from here, Run "python tools/infer.py --weights yolov6s.pt --source img.jpg" while feeding the necessary arguments as needed.