Skip to content
Snippets Groups Projects
Commit 92238c01 authored by Anton Kullberg's avatar Anton Kullberg
Browse files

py: removed tqdm from IMM/Basic tracker

parent 174c9cc3
No related branches found
No related tags found
No related merge requests found
...@@ -52,7 +52,7 @@ class BasicTracker(): ...@@ -52,7 +52,7 @@ class BasicTracker():
numpy.ndarray (nx x nx x K) numpy.ndarray (nx x nx x K)
State error covariance per time. 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 # Calculate prediction error of each measurement
yhat = self.filt.sensor_model['h'](x[:2, k]) yhat = self.filt.sensor_model['h'](x[:2, k])
eps = meas_k-yhat[:, None] eps = meas_k-yhat[:, None]
...@@ -116,7 +116,7 @@ class IMMTracker(): ...@@ -116,7 +116,7 @@ class IMMTracker():
""" """
xm = np.repeat(np.expand_dims(x, -1), self.filt.nmodes, axis=-1) 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) 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 # Get the mixed estimate
xhat, Phat = self.filt.mix(xm[:, k, :], Pm[:, :, k, :]) xhat, Phat = self.filt.mix(xm[:, k, :], Pm[:, :, k, :])
# Calculate prediction error of each measurement # Calculate prediction error of each measurement
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment