diff --git a/docs/tracker/output.mp4 b/docs/tracker/output.mp4 deleted file mode 100644 index 69edeb55..00000000 Binary files a/docs/tracker/output.mp4 and /dev/null differ diff --git a/docs/tracker/tracker_bytetrack_detections.py b/docs/tracker/tracker_bytetrack_detections.py index 2d68b6f9..36ba1de4 100644 --- a/docs/tracker/tracker_bytetrack_detections.py +++ b/docs/tracker/tracker_bytetrack_detections.py @@ -17,7 +17,6 @@ byte_tracker = ByteTrack( match_thresh=0.8, aspect_ratio_thresh=3.0, min_box_area=1.0, - mot20=False, frame_rate=30, ) annotator = BoxAnnotator() diff --git a/docs/tracker/tracker_bytetrack_numpy.py b/docs/tracker/tracker_bytetrack_numpy.py index f5bad81e..81de1ed6 100644 --- a/docs/tracker/tracker_bytetrack_numpy.py +++ b/docs/tracker/tracker_bytetrack_numpy.py @@ -22,7 +22,6 @@ byte_tracker = ByteTrack( match_thresh=0.8, aspect_ratio_thresh=3.0, min_box_area=1.0, - mot20=False, frame_rate=30, ) annotator = BoxAnnotator() diff --git a/docs/tracker/yolov5su.pt b/docs/tracker/yolov5su.pt deleted file mode 100644 index 906325e5..00000000 Binary files a/docs/tracker/yolov5su.pt and /dev/null differ diff --git a/supervision/tracker/byte_tracker/byte_tracker.py b/supervision/tracker/byte_tracker/byte_tracker.py index 3086bb41..bad675e5 100644 --- a/supervision/tracker/byte_tracker/byte_tracker.py +++ b/supervision/tracker/byte_tracker/byte_tracker.py @@ -201,7 +201,6 @@ class ByteTrack: match_thresh=0.8, aspect_ratio_thresh=3.0, min_box_area=1.0, - mot20=False, frame_rate=30, ): @@ -224,7 +223,6 @@ class ByteTrack: self.match_thresh = match_thresh self.aspect_ratio_thresh = aspect_ratio_thresh self.min_box_area = min_box_area - self.mot20 = mot20 self.tracked_stracks = [] # type: list[Strack] self.lost_stracks = [] # type: list[Strack] @@ -308,7 +306,6 @@ class ByteTrack: sink.write_frame(frame) ``` """ - # update tracker tracks = self.update_from_numpy( output_results=detections2boxes(detections=detections), @@ -400,8 +397,8 @@ class ByteTrack: # Predict the current location with KF Strack.multi_predict(strack_pool) dists = matching.iou_distance(strack_pool, detections) - if not self.mot20: - dists = matching.fuse_score(dists, detections) + + dists = matching.fuse_score(dists, detections) matches, u_track, u_detection = matching.linear_assignment( dists, thresh=self.match_thresh ) @@ -454,8 +451,8 @@ class ByteTrack: """Deal with unconfirmed tracks, usually tracks with only one beginning frame""" detections = [detections[i] for i in u_detection] dists = matching.iou_distance(unconfirmed, detections) - if not self.mot20: - dists = matching.fuse_score(dists, detections) + + dists = matching.fuse_score(dists, detections) matches, u_unconfirmed, u_detection = matching.linear_assignment( dists, thresh=0.7 )