This commit is contained in:
2025-02-27 16:04:48 +08:00
parent 096689d6f5
commit 45fff4ac63
2 changed files with 16 additions and 22 deletions

View File

@ -475,9 +475,6 @@ class BoxTracker:
):
evs: list[TrackingEvent]
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):
logger.trace(
"too small bounding boxes; bboxes={}; areas={}",
@ -493,21 +490,17 @@ class BoxTracker:
)
self._confirmed_tracks = self._predict(self._confirmed_tracks, self._params.dt)
self._tentative_tracks = self._predict(self._tentative_tracks, self._params.dt)
c_evs, c_left_m, c_left_bb = (
self._data_associate_and_update( # pylint: disable=E1102
c_evs, c_left_m, c_left_bb = self._data_associate_and_update(
TrackingState.Confirmed, measurements, bounding_boxes
)
)
t_evs, t_left_m, t_left_bb = (
self._data_associate_and_update( # pylint: disable=E1102
t_evs, t_left_m, t_left_bb = self._data_associate_and_update(
TrackingState.Tentative, c_left_m, c_left_bb
)
)
create_c_evs = self._transfer_tentative_to_confirmed(
self._params.survival_steps_threshold
)
# 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_bb,
self._params.dt,

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 (
Literal,
List,
Dict,
TypedDict,
Any,
cast,
Tuple,
Deque,
Dict,
List,
Literal,
Optional,
Sequence,
Deque,
Tuple,
TypedDict,
Union,
cast,
)
import numpy as np
from jaxtyping import Float, Int, Num, Shaped, jaxtyped
try:
from cv2.typing import MatLike
except ImportError: