Function extract_geometric_params

modules/basic_shape_processor.py:82–159

针对特定形状提取几何参数（如平行四边形的倾斜度、圆柱体的顶部高度等）。 返回参数字典，例如 {"size": 0.2, "direction": "south"}

(image: np.ndarray, bbox: list, shape_type: str)

# ======================== 几何参数提取 ========================
def extract_geometric_params(image: np.ndarray, bbox: list, shape_type: str) -> dict:
    """
    针对特定形状提取几何参数（如平行四边形的倾斜度、圆柱体的顶部高度等）。
    返回参数字典，例如 {"size": 0.2, "direction": "south"}
    """
    params = {}
    x1, y1, x2, y2 = map(int, bbox)
    w_box, h_box = x2 - x1, y2 - y1
    
    if w_box <= 0 or h_box <= 0:
        return params

    # 提取 ROI 用于分析
    roi = image[y1:y2, x1:x2]
    gray = cv2.cvtColor(roi, cv2.COLOR_BGR2GRAY)
    
    # 通用预处理：获取轮廓
    _, binary = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
    contours, _ = cv2.findContours(binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
    
    main_cnt = None
    max_area = 0
    for cnt in contours:
        area = cv2.contourArea(cnt)
        if area > max_area:
            max_area = area
            main_cnt = cnt
            
    if main_cnt is None:
        return params

    # 针对不同形状的分析
    if shape_type == "parallelogram":
        # 计算倾斜比例 size (0~1)
        epsilon = 0.02 * cv2.arcLength(main_cnt, True)
        approx = cv2.approxPolyDP(main_cnt, epsilon, True)
        
        if len(approx) == 4:
            pts = approx.reshape(4, 2)
            pts = pts[np.argsort(pts[:, 1])]
            top_pts = pts[:2]
            bottom_pts = pts[2:]
            
            top_pts = top_pts[np.argsort(top_pts[:, 0])]
            bottom_pts = bottom_pts[np.argsort(bottom_pts[:, 0])]
            
            tl, tr = top_pts[0], top_pts[1]
            bl, br = bottom_pts[0], bottom_pts[1]
            
            dx = abs(tl[0] - bl[0])
            size_val = dx / w_box if w_box > 0 else 0.2
            params["size"] = max(0.05, min(0.5, size_val))
            
    elif shape_type == "cylinder":
        params["size"] = max(10, int(w_box * 0.15))
        
    elif shape_type == "triangle":
        epsilon = 0.04 * cv2.arcLength(main_cnt, True)