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 8f1a672a42
1 changed files with 5 additions and 12 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,