search

MuhammadMoinFaisal / YOLOv8-DeepSORT-Object-Tracking

YOLOv8 Object Tracking using PyTorch, OpenCV and DeepSORT
807 stars 222 forks source link

Select a class #23

Closed somamrae closed 1 year ago

somamrae commented 1 year ago

Hi. Thanks for your code. Do you know how to set the desired class? I mean , in which file we can set a class label/number to show in the output. Actually, I'd like to only show "person"s in the output, not all the COCO labels. Thanks

Xbotgo-Justin1 commented 1 year ago

Hello, author! i have the same requirements. pls tell me how to deal with it. Thx!!!

Liuchuang-alt commented 1 year ago

Which platform are you running under linux or windows

somamrae commented 1 year ago

I am running it on Google colab

MuhammadMoinFaisal commented 1 year ago

You can update the predict,py file and add the if statement for example if label==person then only draw bounding boxes around detected objects

smamraee commented 1 year ago

Thanks, And how about the confidence level? How i can modify the confidence level for the detection?

smamraee commented 1 year ago

Hi, I need to save the path for each object. Could you please tell me how I can do it?