From ff577c2ceb4b6aad62b0e4c2e6cbfde6612a47c6 Mon Sep 17 00:00:00 2001 From: Yassine Date: Fri, 13 Oct 2023 16:14:33 -0700 Subject: [PATCH] add lane lines stds --- selfdrive/modeld/fill_model_msg.py | 1 + 1 file changed, 1 insertion(+) diff --git a/selfdrive/modeld/fill_model_msg.py b/selfdrive/modeld/fill_model_msg.py index 9f39fb6f32..72f1e77f93 100644 --- a/selfdrive/modeld/fill_model_msg.py +++ b/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) 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]) + modelV2.laneLineStds = net_output_data['lane_lines_stds'][0,:,0,0].tolist() modelV2.laneLineProbs = net_output_data['lane_lines_prob'][0].tolist() # road edges