From aef9f0c5511acb177cd59f6745e563833f5e0517 Mon Sep 17 00:00:00 2001
From: Anton Kullberg <anton.kullberg@liu.se>
Date: Thu, 11 Nov 2021 17:56:27 +0100
Subject: [PATCH] bug: fixed list comparison bug
---
src/plotters.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/plotters.py b/src/plotters.py
index 0af42b0..bf93c24 100644
--- a/src/plotters.py
+++ b/src/plotters.py
@@ -117,13 +117,13 @@ def plot_result_ex2_24(result):
fig.add_subplot(gs[1:, :])]
Yt = np.hstack(result['Y'])
yx, yy = radar_to_pos(Yt)
-
+ confirmed_id = [track['identity'] for track in result['confirmed_tracks']]
for track in result['tracks']:
x = np.vstack(track['x'])
t = np.hstack(track['t']).flatten()
assoc = np.hstack(track['associations']).flatten()
- if track in result['confirmed_tracks']:
+ if track['identity'] in confirmed_id:
ls = '-'
l = ax[0].plot(t, track['identity']*np.ones(t.shape), ls=ls, markersize=3)[0]
ax[0].plot(assoc, track['identity']*np.ones(assoc.shape), ls='', color=l.get_color(), marker='x', markersize=6)
--
GitLab