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 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226
| 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])
class AdaptiveAirbagController: """自适应气囊控制器""" def __init__(self): self.restraint = ZFAdaptiveRestraint() self.airbag_params = { 'deployment_time': 30, 'max_power': 100, 'min_power': 0 } def update_policy(self, sensor_data: SensorData) -> Dict: """ 更新气囊策略 Args: sensor_data: 传感器数据 Returns: policy: 气囊展开策略 """ occupant_class, confidence = self.restraint.classify_occupant(sensor_data) policy = self.restraint.get_airbag_policy(occupant_class) policy['occupant_class'] = occupant_class.name policy['confidence'] = confidence policy['timestamp'] = time.time() return policy def execute_deployment(self, policy: Dict) -> Dict: """ 执行气囊展开 Args: policy: 展开策略 Returns: result: 展开结果 """ if not policy['enabled']: return { 'deployed': False, 'reason': f"Disabled for {policy['occupant_class']}" } power = policy['power'] deployment_time = self.airbag_params['deployment_time'] return { 'deployed': True, 'power': power, 'deployment_time': deployment_time, 'occupant_class': policy['occupant_class'] }
if __name__ == "__main__": import time controller = AdaptiveAirbagController() 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 ) policy = controller.update_policy(sensor_data) print(f"乘员类别: {policy['occupant_class']}") print(f"置信度: {policy['confidence']:.2f}") print(f"气囊策略: enabled={policy['enabled']}, power={policy['power']}%")
|