|
|
@ -240,6 +240,8 @@ class DriverStatus(): |
|
|
|
|
|
|
|
|
|
|
|
self.face_detected = driver_data.faceProb > self.settings._FACE_THRESHOLD |
|
|
|
self.face_detected = driver_data.faceProb > self.settings._FACE_THRESHOLD |
|
|
|
self.pose.roll, self.pose.pitch, self.pose.yaw = face_orientation_from_net(driver_data.faceOrientation, driver_data.facePosition, cal_rpy) |
|
|
|
self.pose.roll, self.pose.pitch, self.pose.yaw = face_orientation_from_net(driver_data.faceOrientation, driver_data.facePosition, cal_rpy) |
|
|
|
|
|
|
|
if self.wheel_on_right: |
|
|
|
|
|
|
|
self.pose.yaw *= -1 |
|
|
|
self.pose.pitch_std = driver_data.faceOrientationStd[0] |
|
|
|
self.pose.pitch_std = driver_data.faceOrientationStd[0] |
|
|
|
self.pose.yaw_std = driver_data.faceOrientationStd[1] |
|
|
|
self.pose.yaw_std = driver_data.faceOrientationStd[1] |
|
|
|
model_std_max = max(self.pose.pitch_std, self.pose.yaw_std) |
|
|
|
model_std_max = max(self.pose.pitch_std, self.pose.yaw_std) |
|
|
|