1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90
| class DMSApplication: """ DMS应用:使用STURDeCAM57进行驾驶员监控 """ def __init__(self): self.camera = STURDeCAM57() self.face_detector = FaceDetector() self.eye_tracker = EyeTracker() self.gaze_estimator = GazeEstimator() self.distraction_detector = DistractionDetector() def process_frame(self, frame: np.ndarray, vehicle_data: dict) -> dict: """ 处理单帧DMS Args: frame: 图像帧 vehicle_data: 车辆数据 Returns: result: DMS结果 """ faces = self.face_detector.detect(frame) if not faces: return { 'driver_present': False, 'distraction': None, 'gaze': None } eyes = self.eye_tracker.track(frame, faces[0]) gaze = self.gaze_estimator.estimate(eyes) distraction = self.distraction_detector.detect(gaze, vehicle_data) return { 'driver_present': True, 'face_box': faces[0], 'eyes': eyes, 'gaze': gaze, 'distraction': distraction }
class FaceDetector: """人脸检测器""" def detect(self, frame: np.ndarray) -> list: return [(400, 300, 600, 500)]
class EyeTracker: """眼动追踪器""" def track(self, frame: np.ndarray, face: tuple) -> dict: return { 'left_eye': (450, 380), 'right_eye': (550, 380), 'eye_openness': 0.8 }
class GazeEstimator: """视线估计器""" def estimate(self, eyes: dict) -> dict: return { 'pitch': 0.1, 'yaw': -0.2, 'gaze_point': (800, 400) }
class DistractionDetector: """分心检测器""" def detect(self, gaze: dict, vehicle_data: dict) -> dict: is_distracted = abs(gaze['yaw']) > 0.5 return { 'is_distracted': is_distracted, 'distraction_type': 'side' if is_distracted else 'none', 'confidence': 0.95 }
|