crosstyan/SimpleTracker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
crosstyan
2025-02-27 16:04:48 +08:00
parent 096689d6f5
commit 45fff4ac63
2 changed files with 16 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
app/tracker/bboxes_tracker.py
View File

@ -475,9 +475,6 @@ class BoxTracker:
): ):
evs: list[TrackingEvent] evs: list[TrackingEvent]
areas = bounding_boxes_area(bounding_boxes, self._bounding_boxes_format) areas = bounding_boxes_area(bounding_boxes, self._bounding_boxes_format)
# 10 x 10 is too small for a normal bounding box
# filter out
# TODO: use area as gating threshold
if any(areas <= AREA_FILTER_THRESHOLD): if any(areas <= AREA_FILTER_THRESHOLD):
logger.trace( logger.trace(
"too small bounding boxes; bboxes={}; areas={}", "too small bounding boxes; bboxes={}; areas={}",
@ -493,21 +490,17 @@ class BoxTracker:
) )
self._confirmed_tracks = self._predict(self._confirmed_tracks, self._params.dt) self._confirmed_tracks = self._predict(self._confirmed_tracks, self._params.dt)
self._tentative_tracks = self._predict(self._tentative_tracks, self._params.dt) self._tentative_tracks = self._predict(self._tentative_tracks, self._params.dt)
c_evs, c_left_m, c_left_bb = ( c_evs, c_left_m, c_left_bb = self._data_associate_and_update(
self._data_associate_and_update( # pylint: disable=E1102 TrackingState.Confirmed, measurements, bounding_boxes
TrackingState.Confirmed, measurements, bounding_boxes
)
) )
t_evs, t_left_m, t_left_bb = ( t_evs, t_left_m, t_left_bb = self._data_associate_and_update(
self._data_associate_and_update( # pylint: disable=E1102 TrackingState.Tentative, c_left_m, c_left_bb
TrackingState.Tentative, c_left_m, c_left_bb
)
) )
create_c_evs = self._transfer_tentative_to_confirmed( create_c_evs = self._transfer_tentative_to_confirmed(
self._params.survival_steps_threshold self._params.survival_steps_threshold
) )
# target initialize # target initialize
create_t_evs = self._tracks_from_past_measurements( # pylint: disable=E1102 create_t_evs = self._tracks_from_past_measurements(
t_left_m, t_left_m,
t_left_bb, t_left_bb,
self._params.dt, self._params.dt,

21
app/typing/__init__.py
View File

@ -1,19 +1,20 @@
from typing import Any, Union
import numpy as np
from jaxtyping import Float, Int, Shaped, Num, jaxtyped
from typing import ( from typing import (
Literal,
List,
Dict,
TypedDict,
Any, Any,
cast, Deque,
Tuple, Dict,
List,
Literal,
Optional, Optional,
Sequence, Sequence,
Deque, Tuple,
TypedDict,
Union,
cast,
) )
import numpy as np
from jaxtyping import Float, Int, Num, Shaped, jaxtyped
try: try:
from cv2.typing import MatLike from cv2.typing import MatLike
except ImportError: except ImportError: