diff --git a/src/plotters.py b/src/plotters.py index 0af42b03c9c4a552371e6d7c734b7d5cf4822bf8..bf93c24b36032c94acece440f9567bac724f50c8 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)