filter out detections with zero area.

This commit is contained in:
Raif Olson committed 2024-05-08 11:07:14 -04:00
1 parent d7b6af84e6
commit c47b0ce990
1 file changed
+7
+7
View File
@@ -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