mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-08-21 08:14:00 +08:00
Revert fullframe DM model (#24812)
* Revert "fullframe DM: flip RHD yaw to use matching thresholds" This reverts commit ce7daabc8847d18ba46e5d1879f5a6958d04ccc7. * Revert "fullframe DM model (#24762)" This reverts commit 817be81fb19004f4873881f6b29dcdfffbe7e3a8. * revert cereal old-commit-hash: c646eeee0ac54925db5afc51b95c5d869d6dba68
This commit is contained in:
@@ -17,19 +17,19 @@ INVISIBLE_SECONDS_TO_ORANGE = dm_settings._AWARENESS_TIME - dm_settings._AWARENE
|
||||
INVISIBLE_SECONDS_TO_RED = dm_settings._AWARENESS_TIME + 1
|
||||
|
||||
def make_msg(face_detected, distracted=False, model_uncertain=False):
|
||||
ds = log.DriverStateV2.new_message()
|
||||
ds.leftDriverData.faceOrientation = [0., 0., 0.]
|
||||
ds.leftDriverData.facePosition = [0., 0.]
|
||||
ds.leftDriverData.faceProb = 1. * face_detected
|
||||
ds.leftDriverData.leftEyeProb = 1.
|
||||
ds.leftDriverData.rightEyeProb = 1.
|
||||
ds.leftDriverData.leftBlinkProb = 1. * distracted
|
||||
ds.leftDriverData.rightBlinkProb = 1. * distracted
|
||||
ds.leftDriverData.faceOrientationStd = [1.*model_uncertain, 1.*model_uncertain, 1.*model_uncertain]
|
||||
ds.leftDriverData.facePositionStd = [1.*model_uncertain, 1.*model_uncertain]
|
||||
ds = log.DriverState.new_message()
|
||||
ds.faceOrientation = [0., 0., 0.]
|
||||
ds.facePosition = [0., 0.]
|
||||
ds.faceProb = 1. * face_detected
|
||||
ds.leftEyeProb = 1.
|
||||
ds.rightEyeProb = 1.
|
||||
ds.leftBlinkProb = 1. * distracted
|
||||
ds.rightBlinkProb = 1. * distracted
|
||||
ds.faceOrientationStd = [1.*model_uncertain, 1.*model_uncertain, 1.*model_uncertain]
|
||||
ds.facePositionStd = [1.*model_uncertain, 1.*model_uncertain]
|
||||
# TODO: test both separately when e2e is used
|
||||
ds.leftDriverData.readyProb = [0., 0., 0., 0.]
|
||||
ds.leftDriverData.notReadyProb = [0., 0.]
|
||||
ds.readyProb = [0., 0., 0., 0.]
|
||||
ds.notReadyProb = [0., 0.]
|
||||
return ds
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user