diff --git a/src/trackers.py b/src/trackers.py index a13b3592da05fab2c5efaa82bffa114181750e25..07a8960d0f0386aa362175a3fc74fcdaa452bccc 100644 --- a/src/trackers.py +++ b/src/trackers.py @@ -52,7 +52,7 @@ class BasicTracker(): numpy.ndarray (nx x nx x K) State error covariance per time. """ - for k, meas_k in tqdm.tqdm(enumerate(Y), desc="Evaluating measurements: ", total=len(Y)): + for k, meas_k in enumerate(Y): # Calculate prediction error of each measurement yhat = self.filt.sensor_model['h'](x[:2, k]) eps = meas_k-yhat[:, None] @@ -116,7 +116,7 @@ class IMMTracker(): """ xm = np.repeat(np.expand_dims(x, -1), self.filt.nmodes, axis=-1) Pm = np.repeat(np.expand_dims(P, -1), self.filt.nmodes, axis=-1) - for k, meas_k in tqdm.tqdm(enumerate(Y), desc="IMM evaluating measurements: ", total=len(Y)): + for k, meas_k in enumerate(Y): # Get the mixed estimate xhat, Phat = self.filt.mix(xm[:, k, :], Pm[:, :, k, :]) # Calculate prediction error of each measurement