Closed VYRION-Ai closed 1 year ago
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.
@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:`
Thanks!
How to do detection in only selected classes , not all my classes