Object tracking on a custom dataset

Question

I'm new on computer vision field , I work on a project to detect and track trashs. I use YOLO v5 to detect objects now I want to count each object that is present on the Video . I want some suggestion if there are some models that we can use on my own dataset.

recommendation are off-topic. please review [help/on-topic] – Christoph Rackwitz Jul 12 '22 at 10:23 — Christoph Rackwitz, Jul 12 '22 at 10:23

Mike B · Answer 1 · 2022-07-11T14:36:01.207

0

This repo contains all you need for tracking and counting objects: https://github.com/mikel-brostrom/Yolov5_StrongSORT_OSNet. If you already have a set of weights you could start tracking by:

 python track.py --yolo-weights /path/to/your/weights.pt

Adding the counting functionality should be straight forward

edited Jul 11 '22 at 14:36

answered Jul 11 '22 at 13:43

Mike B

2,136
2
12
31

thank you, it work for me but results not good it count 100 time the same object , How I can I perform the result thanks in advance – Ahmed Lafdhal Jul 15 '22 at 14:23

Object tracking on a custom dataset

1 Answers1