From c47b0ce9907fcc776a6a803d97acf9be7bd7770a Mon Sep 17 00:00:00 2001 From: Raif Olson Date: Wed, 8 May 2024 11:07:14 -0400 Subject: [PATCH] filter out detections with zero area. --- supervision/tracker/byte_tracker/core.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/supervision/tracker/byte_tracker/core.py b/supervision/tracker/byte_tracker/core.py index ce3bbbbf..132bf391 100644 --- a/supervision/tracker/byte_tracker/core.py +++ b/supervision/tracker/byte_tracker/core.py @@ -362,6 +362,13 @@ class ByteTrack: scores = tensors[:, 4] bboxes = tensors[:, :4] + bbox_areas = (bboxes[:, 2] - bboxes[:, 0]) * (bboxes[:, 3] - bboxes[:, 1]) + valid_box_inds = bbox_areas > 0 + + class_ids = class_ids[valid_box_inds] + scores = scores[valid_box_inds] + bboxes = bboxes[valid_box_inds] + remain_inds = scores > self.track_activation_threshold inds_low = scores > 0.1 inds_high = scores < self.track_activation_threshold