add lane lines stds

pull/30273/head
Yassine 2 years ago
parent 51fcff02d0
commit ff577c2ceb
  1. 1
      selfdrive/modeld/fill_model_msg.py

@ -53,6 +53,7 @@ def fill_model_msg(msg: capnp._DynamicStructBuilder, net_output_data: Dict[str,
modelV2.init('laneLines', 4) modelV2.init('laneLines', 4)
for i in range(4): for i in range(4):
fill_xyzt(modelV2.laneLines[i], T_IDXS, np.array(X_IDXS), net_output_data['lane_lines'][0,i,:,0], net_output_data['lane_lines'][0,i,:,1]) fill_xyzt(modelV2.laneLines[i], T_IDXS, np.array(X_IDXS), net_output_data['lane_lines'][0,i,:,0], net_output_data['lane_lines'][0,i,:,1])
modelV2.laneLineStds = net_output_data['lane_lines_stds'][0,:,0,0].tolist()
modelV2.laneLineProbs = net_output_data['lane_lines_prob'][0].tolist() modelV2.laneLineProbs = net_output_data['lane_lines_prob'][0].tolist()
# road edges # road edges

Loading…
Cancel
Save