HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.2.74 add fuse_score=True BoT-SORT and ByteTrack arg (#14965)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Laughing 2024-08-06 17:29:40 +08:00 • committed by GitHub
parent 8771f923c0
commit 9f59331854
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 8 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

5
ultralytics/trackers/bot_sort.py
View file

@ -179,9 +179,8 @@ class BOTSORT(BYTETracker):
dists = matching.iou_distance(tracks, detections)
dists_mask = dists > self.proximity_thresh
# TODO: mot20
# if not self.args.mot20:
dists = matching.fuse_score(dists, detections)
if self.args.fuse_score:
dists = matching.fuse_score(dists, detections)
if self.args.with_reid and self.encoder is not None:
emb_dists = matching.embedding_distance(tracks, detections) / 2.0