Heatmaps bug fix (#17634)

2024-11-19 16:40:42 +05:00 · 2024-11-19 16:40:42 +05:00 · bde3105813
commit bde3105813
parent 35b6fed814
2 changed files with 3 additions and 3 deletions
--- a/ultralytics/solutions/object_counter.py
+++ b/ultralytics/solutions/object_counter.py
@ -63,7 +63,7 @@ class ObjectCounter(BaseSolution):
            >>> track_id = 1
            >>> prev_position = (120, 220)
            >>> cls = 0
-            >>> counter.count_objects(track_line, box, track_id, prev_position, cls)
+            >>> counter.count_objects(current_centroid, track_id, prev_position, cls)
        """
        if prev_position is None or track_id in self.counted_ids:
            return