search

MuhammadMoinFaisal / YOLOv8-DeepSORT-Object-Tracking

YOLOv8 Object Tracking using PyTorch, OpenCV and DeepSORT
761 stars 209 forks source link

Configure the Classes to detect from the model #28

Open MahenderAutonomo opened 1 year ago

MahenderAutonomo commented 1 year ago

I am trying to configure and filter the detection classes but still even after applying the classes filter it is still detecting all the classes.

leandromarcomini commented 1 year ago

How are you applying the filter? I'm still looking for places where I can set the detection classes.

Monkez commented 1 year ago

I modify the deep_sort.py file at line 32 like this to detect only person detections = [Detection(bbox_tlwh[i], conf, features[i],oid) for i, (conf,oid) in enumerate(zip(confidences,oids)) if conf > self.min_confidence and oid==0]