Function _merge_nearby_lines

modules/basic_shape_processor.py:708–753

合并相近的平行线段，减少冗余 Args: lines: 线段列表，格式为 [(y, x1, x2), ...] 或 [(x, y1, y2), ...] threshold: 合并阈值，位置差小于此值的线段会被合并 Returns: 合并后的线段列表

(lines, threshold=10)

# ======================== CV矩形检测优化辅助函数 ========================
def _merge_nearby_lines(lines, threshold=10):
    """
    合并相近的平行线段，减少冗余
    
    Args:
        lines: 线段列表，格式为 [(y, x1, x2), ...] 或 [(x, y1, y2), ...]
        threshold: 合并阈值，位置差小于此值的线段会被合并
        
    Returns:
        合并后的线段列表
    """
    if not lines:
        return []
    
    merged = []
    used = set()
    
    for i, line in enumerate(lines):
        if i in used:
            continue
        
        pos, start, end = line  # y/x, x1/y1, x2/y2
        # 找到所有相近的线段
        group_pos = [pos]
        group_start = [start]
        group_end = [end]
        
        for j, other in enumerate(lines[i+1:], i+1):
            if j in used:
                continue
            o_pos, o_start, o_end = other
            if abs(o_pos - pos) < threshold:
                group_pos.append(o_pos)
                group_start.append(o_start)
                group_end.append(o_end)
                used.add(j)
        
        # 合并为一条线
        merged.append((
            int(np.mean(group_pos)),
            min(group_start),
            max(group_end)
        ))
        used.add(i)
    
    return merged


# ======================== CV结果验证 ========================