webrtc-yolov10n

Running on T4

freddyaboulton HF Staff commited on Sep 25, 2024

Commit

e38752e

1 Parent(s): 8e0c0f6

index

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -68,7 +68,7 @@ class YOLOv10:
             return [], [], []
         # Get the class with the highest confidence
-        class_ids = np.argmax(predictions[:, 4:], axis=1)
         # Get bounding boxes for each object
         boxes = self.extract_boxes(predictions)

             return [], [], []
         # Get the class with the highest confidence
+        class_ids = predictions[:, 5].astype(int)
         # Get bounding boxes for each object
         boxes = self.extract_boxes(predictions)