Add https://youtu.be/uDrn9QZJ2lk and object_counting.py, ai_gym.py updates (#9265)

2024-03-24 22:22:17 +05:00 · 2024-03-24 22:22:17 +05:00 · dcb953bc41
commit dcb953bc41
parent 281e07d7e6
5 changed files with 39 additions and 11 deletions
--- a/ultralytics/solutions/object_counter.py
+++ b/ultralytics/solutions/object_counter.py
@ -171,7 +171,7 @@ class ObjectCounter:
            # Extract tracks
            for box, track_id, cls in zip(boxes, track_ids, clss):
                # Draw bounding box
-                self.annotator.box_label(box, label=f"{track_id}:{self.names[cls]}", color=colors(int(cls), True))
+                self.annotator.box_label(box, label=f"{track_id}:{self.names[cls]}", color=colors(int(track_id), True))

                # Draw Tracks
                track_line = self.track_history[track_id]