diff --git a/spatialyze/video_processor/stages/tracking_2d/strongsort.py b/spatialyze/video_processor/stages/tracking_2d/strongsort.py index b7c4f77..cdd240a 100644 --- a/spatialyze/video_processor/stages/tracking_2d/strongsort.py +++ b/spatialyze/video_processor/stages/tracking_2d/strongsort.py @@ -1,6 +1,6 @@ +import time from pathlib import Path from typing import Literal -import time import torch @@ -43,7 +43,7 @@ def _run(self, payload: "Payload"): names = detections[0][1] assert names is not None - + frame_process_time = [] for (dets, _, dids), ts in StrongSORT.tqdm(zip(detections, trackings)): @@ -69,7 +69,7 @@ def _run(self, payload: "Payload"): for before, after in zip(trajectory[:-1], trajectory[1:]): before.next = after after.prev = before - + self._benchmark.append( { "name": payload.video.videofile,