Skip to content

Commit 152c11f

Browse files
update FPS based on mps
1 parent 38db37b commit 152c11f

File tree

2 files changed

+12
-10
lines changed

2 files changed

+12
-10
lines changed

.github/workflows/benchmark.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -86,10 +86,10 @@ jobs:
8686
tracker_fps["deepocsort"]=12
8787
tracker_fps["bytetrack"]=720
8888
tracker_fps["ocsort"]=890
89-
tracker_fps["strongsort"]=17
90-
tracker_fps["botsort"]=46
89+
tracker_fps["strongsort"]=11
90+
tracker_fps["botsort"]=12
9191
tracker_fps["hybridsort"]=25
92-
tracker_fps["boosttrack"]=25
92+
tracker_fps["boosttrack"]=13
9393
tracker_fps["sfsort"]="6000"
9494
fps=${tracker_fps[$TRACKER_NAME]}
9595
fi

boxmot/utils/timing.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -169,17 +169,19 @@ def log(msg):
169169
reid_fps = fps_from_avg(reid_avg)
170170
log(f"{'ReID':<20} | <blue>{reid_total:<12.1f}</blue> | <blue>{reid_avg:<12.2f}</blue> | <blue>{reid_fps:<10.1f}</blue> | {pct(reid_total):<12.1f}")
171171

172-
# Show association/track in both modes (since we now track association in batch mode too)
172+
# Show association/track in both modes
173173
if track_total > 0:
174174
assoc_avg = assoc_time / frames if frames > 0 else 0
175175
assoc_fps = fps_from_avg(assoc_avg)
176176
log(f"{'Association':<20} | <blue>{assoc_time:<12.1f}</blue> | <blue>{assoc_avg:<12.2f}</blue> | <blue>{assoc_fps:<10.1f}</blue> | {pct(assoc_time):<12.1f}")
177-
178-
if not is_batch_mode:
179-
# In real-time mode, also show track total (which includes reid + assoc)
180-
track_avg = track_total / frames if frames > 0 else 0
181-
track_fps = fps_from_avg(track_avg)
182-
log(f"<bold>{'Track (total)':<20}</bold> | <cyan>{track_total:<12.1f}</cyan> | <cyan>{track_avg:<12.2f}</cyan> | <cyan>{track_fps:<10.1f}</cyan> | {pct(track_total):<12.1f}")
177+
178+
tracking_total = track_total if not is_batch_mode else (reid_total + track_total)
179+
tracking_avg = tracking_total / frames if frames > 0 else 0
180+
tracking_fps = fps_from_avg(tracking_avg)
181+
log(
182+
f"<bold>{'Tracking (total)':<20}</bold> | <cyan>{tracking_total:<12.1f}</cyan> | "
183+
f"<cyan>{tracking_avg:<12.2f}</cyan> | <cyan>{tracking_fps:<10.1f}</cyan> | {pct(tracking_total):<12.1f}"
184+
)
183185

184186
log("<blue>" + "-" * 105 + "</blue>")
185187

0 commit comments

Comments
 (0)