diff --git a/src/trackers.py b/src/trackers.py index 07a8960d0f0386aa362175a3fc74fcdaa452bccc..119179cc3815161e8e3332b005d095f59ce07947 100644 --- a/src/trackers.py +++ b/src/trackers.py @@ -215,7 +215,7 @@ class GNN(): association_matrix[validation_matrix[:, ti], ti] = np.log(track['filt'].sensor_model['PD']*py/(1-track['filt'].sensor_model['PD'])) # PG assumed = 1 return association_matrix, validation_matrix - def _update_track(self, meas, track): + def update_track(self, meas, track): """Handles the update of a certain track with the given measurement(s). Modifies the track in-place! @@ -418,7 +418,7 @@ class JPDA(): likelihood_matrix[ti, np.where(validation_matrix[ti, 1:])[0]+1] = track['filt'].sensor_model['PD']*py return likelihood_matrix, validation_matrix - def _update_track(self, meas, track, association_probability): + def update_track(self, meas, track, association_probability): """Handles the update of a certain track with the given measurement(s). Modifies the track in-place! @@ -680,7 +680,7 @@ class MHT(): validation_matrix[:, ti] = self.gater.gate(track['x'][-1], track['P'][-1], meas) return validation_matrix - def _update_track(self, meas, track): + def update_track(self, meas, track): """Handles the update of a certain track with the given measurement(s). Modifies the track in-place!