rect_box 计算分数

Test/test01.py

@@ -1,500 +1,32 @@
-def validate_rule_ranges(data: dict, path: str = "") -> (bool, str):
-    """
-    递归检查配置中看起来像评分规则的列表。
-    规则：
-    1. 列表中的元素必须包含 min, max, deduction。
-    2. 第一个元素的 min 必须为 0。
-    3. 当前元素的 min 必须等于上一个元素的 max。
-    """
-    if isinstance(data, dict):
-        for key, value in data.items():
-            is_valid, error = validate_rule_ranges(value, path=f"{path}.{key}" if path else key)
-            if not is_valid:
-                return False, error
-
-    elif isinstance(data, list):
-        # 检查这是否是一个规则列表（通过检查第一个元素是否包含特定key）
-        if len(data) > 0 and isinstance(data[0], dict) and "min" in data[0] and "max" in data[0] and "deduction" in \
-                data[0]:
-            current_path = path
-
-            # 1. 检查首项 min 是否为 0
-            first_min = data[0].get("min")
-            if first_min != 0:
-                return False, f"范围错误 [{current_path}]: 第一个区间的 min 必须为 0，当前为 {first_min}"
-
-            # 2. 检查连续性
-            prev_max = None
-            for idx, rule in enumerate(data):
-                current_min = rule.get("min")
-                current_max = rule.get("max")
-
-                # 处理 inf 字符串
-                if current_max == "inf":
-                    current_max_val = float("inf")
-                else:
-                    try:
-                        current_max_val = float(current_max)
-                    except (ValueError, TypeError):
-                        return False, f"数值错误 [{current_path}]: max 值 '{current_max}' 无效"
-
-                try:
-                    current_min_val = float(current_min)
-                except (ValueError, TypeError):
-                    return False, f"数值错误 [{current_path}]: min 值 '{current_min}' 无效"
-
-                # 检查连续性 (除了第一个元素)
-                if idx > 0:
-                    # 使用一个微小的容差处理浮点数比较，或者直接相等
-                    if prev_max == "inf":
-                        return False, f"逻辑错误 [{current_path}]: 'inf' 只能出现在最后一个区间的 max"
-
-                    prev_max_val = float(prev_max) if prev_max != "inf" else float("inf")
-
-                    if current_min_val != prev_max_val:
-                        return False, f"不连续错误 [{current_path}]: 第 {idx + 1} 项的 min ({current_min}) 不等于上一项的 max ({prev_max})"
-
-                prev_max = current_max
-
-    return True, "验证通过"
-
-
-if __name__ == '__main__':
-    new_config = {
-        "base_score": 10,
-        "corner": {
-            "rules": {
-                "wear_area": [
-                    {
-                        "min": 0,
-                        "max": 0.05,
-                        "deduction": -0.1
-                    },
-                    {
-                        "min": 0.05,
-                        "max": 0.1,
-                        "deduction": -0.5
-                    },
-                    {
-                        "min": 0.1,
-                        "max": 0.25,
-                        "deduction": -1.5
-                    },
-                    {
-                        "min": 0.25,
-                        "max": 0.5,
-                        "deduction": -3
-                    },
-                    {
-                        "min": 0.5,
-                        "max": "inf",
-                        "deduction": -5
-                    }
-                ],
-                "loss_area": [
-                    {
-                        "min": 0,
-                        "max": 0.05,
-                        "deduction": -0.1
-                    },
-                    {
-                        "min": 0.05,
-                        "max": 0.1,
-                        "deduction": -0.5
-                    },
-                    {
-                        "min": 0.1,
-                        "max": 0.25,
-                        "deduction": -1.5
-                    },
-                    {
-                        "min": 0.25,
-                        "max": 0.5,
-                        "deduction": -3
-                    },
-                    {
-                        "min": 0.5,
-                        "max": "inf",
-                        "deduction": -5
-                    }
-                ]
-            },
-            "front_weights": {
-                "wear_area": 0.3,
-                "loss_area": 0.7
-            },
-            "back_weights": {
-                "wear_area": 0.3,
-                "loss_area": 0.7
-            },
-            "final_weights": {
-                "front": 0.7,
-                "back": 0.3
-            }
-        },
-        "edge": {
-            "rules": {
-                "wear_area": [
-                    {
-                        "min": 0,
-                        "max": 0.05,
-                        "deduction": -0.1
-                    },
-                    {
-                        "min": 0.05,
-                        "max": 0.1,
-                        "deduction": -0.5
-                    },
-                    {
-                        "min": 0.1,
-                        "max": 0.25,
-                        "deduction": -1.5
-                    },
-                    {
-                        "min": 0.25,
-                        "max": 0.5,
-                        "deduction": -3
-                    },
-                    {
-                        "min": 0.5,
-                        "max": "inf",
-                        "deduction": -5
-                    }
-                ],
-                "loss_area": [
-                    {
-                        "min": 0,
-                        "max": 0.05,
-                        "deduction": -0.1
-                    },
-                    {
-                        "min": 0.05,
-                        "max": 0.1,
-                        "deduction": -0.5
-                    },
-                    {
-                        "min": 0.1,
-                        "max": 0.25,
-                        "deduction": -1.5
-                    },
-                    {
-                        "min": 0.25,
-                        "max": 0.5,
-                        "deduction": -3
-                    },
-                    {
-                        "min": 0.5,
-                        "max": "inf",
-                        "deduction": -5
-                    }
-                ]
-            },
-            "front_weights": {
-                "wear_area": 0.4,
-                "loss_area": 0.6
-            },
-            "back_weights": {
-                "wear_area": 0.4,
-                "loss_area": 0.6
-            },
-            "final_weights": {
-                "front": 0.7,
-                "back": 0.3
-            }
-        },
-        "face": {
-            "rules": {
-                "wear_area": [
-                    {
-                        "min": 0,
-                        "max": 0.05,
-                        "deduction": -0.1
-                    },
-                    {
-                        "min": 0.05,
-                        "max": 0.1,
-                        "deduction": -0.5
-                    },
-                    {
-                        "min": 0.1,
-                        "max": 0.25,
-                        "deduction": -1.5
-                    },
-                    {
-                        "min": 0.25,
-                        "max": 0.5,
-                        "deduction": -3
-                    },
-                    {
-                        "min": 0.5,
-                        "max": "inf",
-                        "deduction": -5
-                    }
-                ],
-                "pit_area": [
-                    {
-                        "min": 0,
-                        "max": 0.05,
-                        "deduction": -0.1
-                    },
-                    {
-                        "min": 0.05,
-                        "max": 0.1,
-                        "deduction": -0.5
-                    },
-                    {
-                        "min": 0.1,
-                        "max": 0.25,
-                        "deduction": -1.5
-                    },
-                    {
-                        "min": 0.25,
-                        "max": 0.5,
-                        "deduction": -3
-                    },
-                    {
-                        "min": 0.5,
-                        "max": "inf",
-                        "deduction": -5
-                    }
-                ],
-                "stain_area": [
-                    {
-                        "min": 0,
-                        "max": 0.05,
-                        "deduction": -0.1
-                    },
-                    {
-                        "min": 0.05,
-                        "max": 0.1,
-                        "deduction": -0.5
-                    },
-                    {
-                        "min": 0.1,
-                        "max": 0.25,
-                        "deduction": -1.5
-                    },
-                    {
-                        "min": 0.25,
-                        "max": 0.5,
-                        "deduction": -3
-                    },
-                    {
-                        "min": 0.5,
-                        "max": "inf",
-                        "deduction": -5
-                    }
-                ],
-                "scratch_length": [
-                    {
-                        "min": 0,
-                        "max": 1,
-                        "deduction": -0.1
-                    },
-                    {
-                        "min": 1,
-                        "max": 2,
-                        "deduction": -0.5
-                    },
-                    {
-                        "min": 2,
-                        "max": 5,
-                        "deduction": -1
-                    },
-                    {
-                        "min": 5,
-                        "max": 10,
-                        "deduction": -2
-                    },
-                    {
-                        "min": 10,
-                        "max": 20,
-                        "deduction": -3
-                    },
-                    {
-                        "min": 20,
-                        "max": 50,
-                        "deduction": -4
-                    },
-                    {
-                        "min": 50,
-                        "max": "inf",
-                        "deduction": -5
-                    }
-                ]
-            },
-            "coefficients": {
-                "wear_area": 0.25,
-                "scratch_length": 0.25,
-                "dent_area": 0.25,
-                "stain_area": 0.25
-            },
-            "final_weights": {
-                "front": 0.75,
-                "back": 0.25
-            }
-        },
-        "centering": {
-            "front": {
-                "rules": [
-                    {
-                        "min": 0,
-                        "max": 52,
-                        "deduction": 0
-                    },
-                    {
-                        "min": 52,
-                        "max": 55,
-                        "deduction": -0.5
-                    },
-                    {
-                        "min": 55,
-                        "max": 60,
-                        "deduction": -1
-                    },
-                    {
-                        "min": 60,
-                        "max": 62.5,
-                        "deduction": -1.5
-                    },
-                    {
-                        "min": 62.5,
-                        "max": 65,
-                        "deduction": -2
-                    },
-                    {
-                        "min": 65,
-                        "max": 67.5,
-                        "deduction": -2.5
-                    },
-                    {
-                        "min": 67.5,
-                        "max": 70,
-                        "deduction": -3
-                    },
-                    {
-                        "min": 70,
-                        "max": 72.5,
-                        "deduction": -3.5
-                    },
-                    {
-                        "min": 72.5,
-                        "max": 75,
-                        "deduction": -4
-                    },
-                    {
-                        "min": 75,
-                        "max": 77.5,
-                        "deduction": -4.5
-                    },
-                    {
-                        "min": 77.5,
-                        "max": 80,
-                        "deduction": -5
-                    },
-                    {
-                        "min": 80,
-                        "max": 82.5,
-                        "deduction": -5.5
-                    },
-                    {
-                        "min": 82.5,
-                        "max": 85,
-                        "deduction": -6
-                    },
-                    {
-                        "min": 85,
-                        "max": 87.5,
-                        "deduction": -6.5
-                    },
-                    {
-                        "min": 87.5,
-                        "max": 90,
-                        "deduction": -7
-                    },
-                    {
-                        "min": 90,
-                        "max": 92.5,
-                        "deduction": -7.5
-                    },
-                    {
-                        "min": 92.5,
-                        "max": 95,
-                        "deduction": -8
-                    },
-                    {
-                        "min": 95,
-                        "max": 97.5,
-                        "deduction": -8.5
-                    },
-                    {
-                        "min": 97.5,
-                        "max": "inf",
-                        "deduction": -9
-                    }
-                ],
-                "coefficients": {
-                    "horizontal": 1.2,
-                    "vertical": 0.9
-                }
-            },
-            "back": {
-                "rules": [
-                    {
-                        "min": 0,
-                        "max": 60,
-                        "deduction": -0.5
-                    },
-                    {
-                        "min": 60,
-                        "max": 70,
-                        "deduction": -1
-                    },
-                    {
-                        "min": 70,
-                        "max": 75,
-                        "deduction": -1.5
-                    },
-                    {
-                        "min": 75,
-                        "max": 85,
-                        "deduction": -2
-                    },
-                    {
-                        "min": 85,
-                        "max": 95,
-                        "deduction": -2.5
-                    },
-                    {
-                        "min": 95,
-                        "max": "inf",
-                        "deduction": -3
-                    }
-                ],
-                "coefficients": {
-                    "horizontal": 1.2,
-                    "vertical": 0.9
-                }
-            },
-            "final_weights": {
-                "front": 0.75,
-                "back": 0.25
-            }
-        },
-        "card": {
-            "PSA": {
-                "face": 0.35,
-                "corner": 0.3,
-                "edge": 0.1,
-                "center": 0.25
-            },
-            "BGS": {
-                "face": 0.3,
-                "corner": 0.25,
-                "edge": 0.1,
-                "center": 0.25
-            }
-        }
-    }
-    is_range_valid, range_reason = validate_rule_ranges(new_config)
-    print(is_range_valid, range_reason)
+import cv2
+import numpy as np
+
+data = np.array([
+    [
+        282,
+        316
+    ],
+    [
+        2593,
+        316
+    ],
+    [
+        2593,
+        3601
+    ],
+    [
+        282,
+        3601
+    ]
+])
+
+ix, iy, iw, ih = cv2.boundingRect(data)
+
+
+def get_box_corners(x, y, w, h):
+    return [[x, y], [x + w, y], [x + w, y + h], [x, y + h]]
+
+
+print(ix, iy, iw, ih)
+inner_box_corners_int = get_box_corners(ix, iy, iw, ih)
+print(inner_box_corners_int)

app/utils/defect_inference/AnalyzeCenter.py

@@ -14,6 +14,12 @@ def get_points_from_file(json_file):
         data = json.load(f)
     return data['shapes'][0]['points']
 
+def get_rect_from_file(json_file):
+    """辅助函数，从JSON文件加载点"""
+    with open(json_file, 'r') as f:
+        data = json.load(f)
+    return data['shapes'][0]['rect_box']
+
 
 def draw_rotated_bounding_boxes(image_path, inner_box_file, outer_box_file, output_path=None):
     """
@@ -214,9 +220,9 @@ def analyze_centering_rect(inner_points: Union[str, List], outer_points: Union[s
     适用于前端已经修正过的矩形数据。
     """
     if isinstance(inner_points, str):
-        inner_points = get_points_from_file(inner_points)
+        inner_points = get_rect_from_file(inner_points)
     if isinstance(outer_points, str):
-        outer_points = get_points_from_file(outer_points)
+        outer_points = get_rect_from_file(outer_points)
 
     # 转换为 numpy 数组
     inner_contour = np.array(inner_points, dtype=np.int32)

app/utils/defect_inference/arean_anylize_draw.py

@@ -280,7 +280,7 @@ class DefectProcessor:
 
         for defect in json_data['defects']:
             label = defect.get('label', 'unlabeled')
-            points = defect.get('points')
+            points = defect.get('rect_box')
             if not points:
                 continue