x
This commit is contained in:
@ -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
|
||||
TrackingState.Confirmed, measurements, bounding_boxes
|
||||
)
|
||||
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
|
||||
TrackingState.Tentative, c_left_m, c_left_bb
|
||||
)
|
||||
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,
|
||||
|
||||
@ -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:
|
||||
|
||||
Reference in New Issue
Block a user