How to do detection in only selected classes , not all my classes

[VYRION-Ai]

How to do detection in only selected classes , not all my classes

[Tamminhdiep97]

I do a little trick like this in yolov8/utils.py cls_id = 0 is for human only

well, i don't really know anyway other than this, i know that this trick is only work on display on image, the network will still detecting all the classes, so anyone know the proper answer please delight us.

[VYRION-Ai]

@Tamminhdiep97 as you said , just doing filter like this at top class_filter=[0,1,2,3,4,5,6,7]

`    for box, score, class_id in zip(boxes, scores, class_ids):

      if class_id in class_filter:`

[ibaiGorordo]

Thanks!