diff --git a/app/tracker/bboxes_tracker.py b/app/tracker/bboxes_tracker.py index e7618fe..7d5b303 100644 --- a/app/tracker/bboxes_tracker.py +++ b/app/tracker/bboxes_tracker.py @@ -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,