mawei/StarPilot
1
0
Fork 0
mirror of https://github.com/firestar5683/StarPilot.git synced 2026-07-18 15:52:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

sleepProb

Browse Source
This commit is contained in:
firestar5683
2026-07-16 12:33:27 -05:00
parent d935514f78
commit 5a7ec6b154
3 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cereal/libcereal.a
BIN
View File
Binary file not shown.
cereal/log.capnp
+1
View File
@@ -2192,6 +2192,7 @@ struct DriverStateV2 {
rightBlinkProb @8 :Float32;
sunglassesProb @9 :Float32;
phoneProb @13 :Float32;
sleepProb @14 :Float32;
notReadyProbDEPRECATED @12 :List(Float32);
occludedProbDEPRECATED @10 :Float32;
readyProbDEPRECATED @11 :List(Float32);
selfdrive/modeld/dmonitoringmodeld.py
+1 -4
View File
@@ -114,10 +114,7 @@ def fill_driver_data(msg, model_output, suffix):
msg.rightBlinkProb = model_output[f"right_blink_prob_{suffix}"][0, 0].item()
msg.sunglassesProb = model_output[f"sunglasses_prob_{suffix}"][0, 0].item()
msg.phoneProb = model_output[f"using_phone_prob_{suffix}"][0, 0].item()
try:
msg.sleepProb = model_output[f"sleep_prob_{suffix}"][0, 0].item()
except AttributeError:
pass
msg.sleepProb = model_output[f"sleep_prob_{suffix}"][0, 0].item()
def get_driverstate_packet(model_output, frame_id: int, exec_time: float, gpu_exec_time: float):