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 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163
| import numpy as np from typing import Dict, Tuple from dataclasses import dataclass from enum import Enum
class OccupantClass(Enum): """乘员分类""" EMPTY = 0 ADULT_SMALL = 1 ADULT_MEDIUM = 2 ADULT_LARGE = 3 CHILD = 4 CHILD_SEAT = 5 UNKNOWN = 6
@dataclass class SensorData: """传感器数据""" camera: Dict = None seat_pressure: np.ndarray = None seat_weight: float = 0.0 belt_tension: float = 0.0 belt_extension: float = 0.0
class ZFAdaptiveRestraint: """ZF自适应约束系统""" def __init__(self): self.weight_thresholds = { 'empty': 5, 'child': 30, 'small': 50, 'medium': 80, 'large': 80 } self.airbag_policy = { OccupantClass.EMPTY: {'enabled': False, 'power': 0}, OccupantClass.ADULT_SMALL: {'enabled': True, 'power': 60}, OccupantClass.ADULT_MEDIUM: {'enabled': True, 'power': 80}, OccupantClass.ADULT_LARGE: {'enabled': True, 'power': 100}, OccupantClass.CHILD: {'enabled': True, 'power': 40}, OccupantClass.CHILD_SEAT: {'enabled': False, 'power': 0}, OccupantClass.UNKNOWN: {'enabled': True, 'power': 70} } def classify_occupant(self, sensor_data: SensorData) -> Tuple[OccupantClass, float]: """ 分类乘员 Args: sensor_data: 传感器数据 Returns: occupant_class: 乘员类别 confidence: 置信度 """ scores = {} if sensor_data.camera: camera_class = self._classify_from_camera(sensor_data.camera) scores['camera'] = {'class': camera_class, 'weight': 0.4} seat_class = self._classify_from_seat(sensor_data.seat_weight) scores['seat'] = {'class': seat_class, 'weight': 0.35} belt_class = self._classify_from_belt(sensor_data.belt_tension) scores['belt'] = {'class': belt_class, 'weight': 0.25} final_class = self._weighted_fusion(scores) confidence = self._compute_confidence(scores, final_class) return final_class, confidence def _classify_from_camera(self, camera_data: Dict) -> OccupantClass: """摄像头分类""" body_size = camera_data.get('estimated_size', 'medium') if body_size == 'small': return OccupantClass.ADULT_SMALL elif body_size == 'medium': return OccupantClass.ADULT_MEDIUM else: return OccupantClass.ADULT_LARGE def _classify_from_seat(self, weight: float) -> OccupantClass: """座椅传感器分类""" if weight < self.weight_thresholds['empty']: return OccupantClass.EMPTY elif weight < self.weight_thresholds['child']: return OccupantClass.CHILD elif weight < self.weight_thresholds['small']: return OccupantClass.ADULT_SMALL elif weight < self.weight_thresholds['medium']: return OccupantClass.ADULT_MEDIUM else: return OccupantClass.ADULT_LARGE def _classify_from_belt(self, tension: float) -> OccupantClass: """安全带传感器分类""" if tension < 1.0: return OccupantClass.EMPTY else: return OccupantClass.ADULT_MEDIUM def _weighted_fusion(self, scores: Dict) -> OccupantClass: """加权融合""" votes = {} for source, data in scores.items(): cls = data['class'] weight = data['weight'] if cls not in votes: votes[cls] = 0 votes[cls] += weight return max(votes, key=votes.get) def _compute_confidence(self, scores: Dict, final_class: OccupantClass) -> float: """计算置信度""" agreements = sum(1 for s in scores.values() if s['class'] == final_class) return agreements / len(scores) def get_airbag_policy(self, occupant_class: OccupantClass) -> Dict: """获取气囊策略""" return self.airbag_policy.get(occupant_class, self.airbag_policy[OccupantClass.UNKNOWN])
if __name__ == "__main__": restraint = ZFAdaptiveRestraint() sensor_data = SensorData( camera={'estimated_size': 'medium'}, seat_pressure=np.random.rand(16, 16), seat_weight=75.0, belt_tension=5.0, belt_extension=0.8 ) occupant_class, confidence = restraint.classify_occupant(sensor_data) policy = restraint.get_airbag_policy(occupant_class) print(f"乘员类别: {occupant_class.name}") print(f"置信度: {confidence:.2f}") print(f"气囊策略: enabled={policy['enabled']}, power={policy['power']}%")
|