20241217

2024-12-17 17:32:09 +08:00
parent afd033b965
commit 39f94c7bd4
11 changed files with 768 additions and 250 deletions
--- a/tracking/utils/drawtracks.py
+++ b/tracking/utils/drawtracks.py
@ -364,7 +364,10 @@ def drawTrack(tracks, img):
    annotator = TrackAnnotator(img, line_width=2)
    
    for track in tracks:
-        annotator.plotting_track(track.boxes)
+        if isinstance(track, np.ndarray):
+            annotator.plotting_track(track)
+        else:
+            annotator.plotting_track(track.boxes)
    img = annotator.result()    
    
    # pth = save_dir.joinpath(f"{filename}")