import json from typing import List, Dict, Any, Union class CardScorer: """ 它从一个JSON配置文件加载评分规则，并根据输入的卡片数据计算分数。 """ def __init__(self, config_path: str): try: with open(config_path, 'r', encoding='utf-8') as f: self.config = json.load(f) self.base_score = self.config.get("base_score", 10.0) except FileNotFoundError: raise ValueError(f"配置文件未找到: {config_path}") except json.JSONDecodeError: raise ValueError(f"配置文件格式错误: {config_path}") def _get_score_from_tiers(self, value: float, rules: List[List[Any]]) -> float: """ 一个通用的辅助函数，根据分层规则查找值对应的分数。 """ for tier in rules: threshold, score = tier if threshold == "inf" or value < threshold: return float(score) return 0.0 # 如果没有匹配的规则，不扣分 def calculate_defect_score(self, card_defect_type: str, card_aspect: str, defect_data: Dict, is_write_score: bool = False) -> Union[float, dict]: """ 一个通用的缺陷计分函数，用于计算角、边、表面的加权分数。 card_defect_type: 'corner', 'edge', 'face' card_aspect: 为 front或 back is_write_score: 是否将分数写入json并返回 """ if card_defect_type != "corner" and card_defect_type != "edge" and card_defect_type != "face": raise TypeError("calculate_centering_score:card_type 只能为 'corner', 'edge', 'face'") if card_aspect != "front" and card_aspect != "back": raise TypeError("calculate_defect_score:card_type 只能为 front 或 back") aspect_config = self.config[card_defect_type] total_deduction = 0.0 weighted_scores = {} # 1. 计算每种缺陷类型的总扣分 for defect in defect_data['defects']: if defect['defect_type'] != card_defect_type: continue if card_defect_type == 'corner' or card_defect_type == 'edge': if defect['label'] in ['wear', 'wear_and_impact', 'wear_and_stain']: defect_type = "wear_area" elif defect['label'] in ['impact', 'damaged']: defect_type = "loss_area" else: raise TypeError(f"数据缺陷类型不存在: {defect['label']}") else: if defect['label'] in ['scratch']: defect_type = "scratch_length" elif defect['label'] in ['pit']: defect_type = "pit_area" elif defect['label'] in ['stain']: defect_type = "stain_area" else: raise TypeError(f"数据缺陷类型不存在: {defect['label']}") rules = aspect_config['rules'][defect_type] # 对于划痕取长度, 其他取面积 if defect_type == "scratch_length": area_mm = max(defect['width'], defect['height']) else: area_mm = defect['actual_area'] # 累加所有同类型缺陷的扣分 if defect_type not in weighted_scores.keys(): weighted_scores[defect_type] = 0 # 计算单个缺陷扣分 the_score = self._get_score_from_tiers(area_mm, rules) print(f"[{card_defect_type}, {defect_type}]: {area_mm}, {the_score}") weighted_scores[defect_type] += the_score # 将分数写入json if is_write_score: defect['score'] = the_score # 2. 根据权重/系数计算最终扣分 weights = aspect_config.get(f"{card_aspect}_weights") or aspect_config.get("coefficients") if not weights: raise ValueError(f"在配置中未找到 '{card_defect_type}' 的权重/系数") print(weighted_scores) for defect_type, score in weighted_scores.items(): total_deduction += score * weights.get(defect_type, 1.0) final_weights = aspect_config["final_weights"][card_aspect] final_score = total_deduction * final_weights print(f"final weights: {final_weights}, final score: {final_score}") if is_write_score: defect_data[f'{card_aspect}_{card_defect_type}_score'] = final_score return defect_data else: return final_score def calculate_centering_score(self, card_aspect: str, center_data: dict, is_write_score: bool = False) -> Union[float, dict]: """ 计算居中度分数。 card_type 为 front或 back is_write_score: 是否将分数写入json并返回 """ if card_aspect != "front" and card_aspect != "back": raise TypeError("calculate_centering_score:card_type 只能为 front 或 back") centering_config = self.config['centering'][card_aspect] rules = centering_config['rules'] coefficients = centering_config['coefficients'] center_left = center_data['center_result']['center_inference']['center_left'] center_right = center_data['center_result']['center_inference']['center_right'] center_top = center_data['center_result']['center_inference']['center_top'] center_bottom = center_data['center_result']['center_inference']['center_bottom'] # 将比例转换为用于查找规则的单个最大值 h_lookup_val = max(center_left, center_right) v_lookup_val = max(center_top, center_bottom) h_deduction = self._get_score_from_tiers(h_lookup_val, rules) * coefficients['horizontal'] v_deduction = self._get_score_from_tiers(v_lookup_val, rules) * coefficients['vertical'] print(h_deduction, v_deduction) final_weight = self.config['centering']["final_weights"][card_aspect] final_score = (h_deduction + v_deduction) * final_weight print(f"final weight: {final_weight}, final score: {final_score}") if is_write_score: center_data[f'{card_aspect}_score'] = final_score return center_data else: return final_score if __name__ == '__main__': # 1. 初始化评分器，加载规则 scorer = CardScorer(r"C:\Code\ML\Project\CheckCardBoxAndDefectServer\app\core\scoring_config.json") # rulers = scorer.config['corners']['rules']['wear_area'] # # score = scorer._get_score_from_tiers(0.06, rulers) # print(score) # print() # 居中分数 center_data_path = r"C:\Code\ML\Project\CheckCardBoxAndDefectServer\_temp_work\pokemon_card_center-center_result.json" with open(center_data_path, 'r', encoding='utf-8') as f: center_data = json.load(f) center_data = scorer.calculate_centering_score("front", center_data, True) print(center_data) # 边角分数 edge_corner_data_path = r"C:\Code\ML\Project\CheckCardBoxAndDefectServer\_temp_work\pokemon_front_corner_no_reflect_defect-corner_result.json" with open(edge_corner_data_path, 'r', encoding='utf-8') as f: edge_corner_data = json.load(f) corner_data = scorer.calculate_defect_score("corner", 'front', edge_corner_data, True) print(corner_data) score = scorer.calculate_defect_score("edge", 'front', edge_corner_data, True) print(score) # 面分数 face_data_path = r"C:\Code\ML\Project\CheckCardBoxAndDefectServer\_temp_work\pokemon_front_face_no_reflect_defect-face_result.json" with open(face_data_path, 'r', encoding='utf-8') as f: face_data = json.load(f) score = scorer.calculate_defect_score("face", 'front', face_data, True) print(score)