diff --git a/spatialyze/video_processor/stream/strongsort.py b/spatialyze/video_processor/stream/strongsort.py index 10bb3a1..ab15553 100644 --- a/spatialyze/video_processor/stream/strongsort.py +++ b/spatialyze/video_processor/stream/strongsort.py @@ -1,9 +1,9 @@ from dataclasses import dataclass, field from pathlib import Path -import torch import numpy as np import numpy.typing as npt +import torch from ..modules.yolo_tracker.trackers.multi_tracker_zoo import StrongSORT as _StrongSORT from ..modules.yolo_tracker.trackers.multi_tracker_zoo import create_tracker @@ -12,7 +12,7 @@ from ..stages.tracking_2d.tracking_2d import Tracking2DResult from ..types import DetectionId from ..video import Video -from .data_types import Detection2D, Skip, skip +from .data_types import Detection2D, Skip from .reusable import reusable from .stream import Stream diff --git a/spatialyze/video_processor/stream/yolo.py b/spatialyze/video_processor/stream/yolo.py index 24e5633..df32fe7 100644 --- a/spatialyze/video_processor/stream/yolo.py +++ b/spatialyze/video_processor/stream/yolo.py @@ -12,10 +12,10 @@ scale_boxes, ) from ..modules.yolo_tracker.yolov5.utils.torch_utils import select_device -from ..stages.detection_2d.yolo_detection import YoloDetection, class_mapping_to_list +from ..stages.detection_2d.yolo_detection import class_mapping_to_list from ..types import DetectionId from ..video.video import Video -from .data_types import Detection2D, Frame, Skip, skip +from .data_types import Detection2D, Skip, skip from .reusable import reusable from .stream import Stream diff --git a/spatialyze/video_processor/utils/prepare_trajectory.py b/spatialyze/video_processor/utils/prepare_trajectory.py index 3c590cc..9e25a8c 100644 --- a/spatialyze/video_processor/utils/prepare_trajectory.py +++ b/spatialyze/video_processor/utils/prepare_trajectory.py @@ -5,7 +5,9 @@ from ..types import Float3 -def prepare_trajectory(video_name: "str", obj_id: "int", track: "list[TrackingResult]", configs: list[CameraConfig]): +def prepare_trajectory( + video_name: "str", obj_id: "int", track: "list[TrackingResult]", configs: list[CameraConfig] +): timestamps: "list[datetime.datetime]" = [] pairs: "list[Float3]" = [] itemHeadings: "list[float | None]" = [None] * len(track)