Skip to content

Commit

Permalink
style: [CI] format
Browse files Browse the repository at this point in the history
  • Loading branch information
Github Actions Bot committed Oct 17, 2023
1 parent b9ec309 commit 603a7f9
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 5 deletions.
4 changes: 2 additions & 2 deletions spatialyze/video_processor/stream/strongsort.py
Original file line number Diff line number Diff line change
@@ -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
Expand All @@ -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

Expand Down
4 changes: 2 additions & 2 deletions spatialyze/video_processor/stream/yolo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
4 changes: 3 additions & 1 deletion spatialyze/video_processor/utils/prepare_trajectory.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 603a7f9

Please sign in to comment.