mawei/sunnypilot
1
0
Fork 0
mirror of https://github.com/sunnypilot/sunnypilot.git synced 2026-02-28 12:54:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

rerun: fix rp_visualization TypeError (#33749)

Browse Source 
* Fixed iteration over liveTracks in radar visualization by ensuring the data structure is iterable and made the background black

* Requested changes, moved conditional to function call

* Simplified conditional, renamed method
This commit is contained in:
Will
2024-10-08 15:50:55 -06:00
committed by GitHub
parent d72b59832c
commit 8a8d8c2272
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tools/replay/rp_visualization.py
View File

@@ -8,7 +8,7 @@ import cereal.messaging as messaging
from openpilot.common.basedir import BASEDIR
from openpilot.tools.replay.lib.rp_helpers import (UP, rerunColorPalette,
get_blank_lid_overlay,
maybe_update_radar_points, plot_lead,
update_radar_points, plot_lead,
plot_model)
from msgq.visionipc import VisionIpcClient, VisionStreamType
@@ -33,7 +33,8 @@ def visualize(addr):
if sm.recv_frame['radarState']:
plot_lead(sm['radarState'], lid_overlay)
liveTracksTime = sm.logMonoTime['liveTracks']
maybe_update_radar_points(sm['liveTracks'], lid_overlay)
if sm.updated['liveTracks']:
update_radar_points(sm['liveTracks'], lid_overlay)
rr.set_time_nanos("TIMELINE", liveTracksTime)
rr.log("tracks", rr.SegmentationImage(np.flip(np.rot90(lid_overlay, k=-1), axis=1)))