ui.py: add bg to radar points

fix
pull/35666/head
Shane Smiskol 1 month ago
parent f088b4320c
commit 9f9940c5a3
  1. 9
      tools/replay/lib/ui_helpers.py

@ -198,15 +198,12 @@ def maybe_update_radar_points(lt, lid_overlay):
ar_pts = {} ar_pts = {}
for track in lt: for track in lt:
ar_pts[track.trackId] = [track.dRel, track.yRel, track.vRel, track.aRel] ar_pts[track.trackId] = [track.dRel, track.yRel, track.vRel, track.aRel]
for ids, pt in ar_pts.items(): for pt in ar_pts.values():
# negative here since radar is left positive # negative here since radar is left positive
px, py = to_topdown_pt(pt[0], -pt[1]) px, py = to_topdown_pt(pt[0], -pt[1])
if px != -1: if px != -1:
color = 255 lid_overlay[px - 4:px + 4, py - 4:py + 4] = 0
if int(ids) == 1: lid_overlay[px - 2:px + 2, py - 2:py + 2] = 255
lid_overlay[px - 2:px + 2, py - 10:py + 10] = 100
else:
lid_overlay[px - 2:px + 2, py - 2:py + 2] = color
def get_blank_lid_overlay(UP): def get_blank_lid_overlay(UP):
lid_overlay = np.zeros((UP.lidar_x, UP.lidar_y), 'uint8') lid_overlay = np.zeros((UP.lidar_x, UP.lidar_y), 'uint8')

Loading…
Cancel
Save