allow all classes to be detected

SkalskiP · Feb 14, 2023 · f19a38f · f19a38f
1 parent 123b8b1
commit f19a38f
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -57,7 +57,6 @@ def main():
 
         result = model(frame, agnostic_nms=True)[0]
         detections = sv.Detections.from_yolov8(result)
-        detections = detections[detections.class_id == 47]
         labels = [
             f"{model.model.names[class_id]} {confidence:0.2f}"
             for _, confidence, class_id, _