Python中如何使用OpenCV去除图片底部的横线？

import cv2
import numpy as np

def remove_bottom_lines(image_path, output_path, line_thickness=2, margin=10):
    """
    去除图片底部的横线
    
    参数:
        image_path: 输入图片路径
        output_path: 输出图片路径
        line_thickness: 检测线条的厚度阈值
        margin: 从底部开始检测的区域高度
    """
    # 读取图片
    img = cv2.imread(image_path)
    if img is None:
        raise ValueError("无法读取图片，请检查路径")
    
    # 转换为灰度图
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    
    # 获取图片尺寸
    height, width = gray.shape
    
    # 提取底部区域（最后margin行）
    bottom_region = gray[height - margin:, :]
    
    # 使用Canny边缘检测
    edges = cv2.Canny(bottom_region, 50, 150, apertureSize=3)
    
    # 使用霍夫变换检测直线
    lines = cv2.HoughLinesP(edges, 1, np.pi/180, threshold=50,
                           minLineLength=width*0.3, maxLineGap=10)
    
    # 如果没有检测到直线，直接保存原图
    if lines is None:
        cv2.imwrite(output_path, img)
        print("未检测到横线")
        return img
    
    # 创建掩码
    mask = np.ones_like(img) * 255
    
    # 标记检测到的横线区域
    for line in lines:
        x1, y1, x2, y2 = line[0]
        # 将局部坐标转换为全局坐标
        y1 += height - margin
        y2 += height - margin
        
        # 计算线条角度（只处理接近水平的线）
        angle = np.abs(np.arctan2(y2 - y1, x2 - x1) * 180 / np.pi)
        if angle < 10 or angle > 170:  # 接近水平线
            # 扩展线条区域
            y_center = (y1 + y2) // 2
            top = max(0, y_center - line_thickness)
            bottom = min(height, y_center + line_thickness)
            mask[top:bottom, :] = 0
    
    # 使用inpainting修复被标记的区域
    mask_gray = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)
    result = cv2.inpaint(img, mask_gray, inpaintRadius=3, flags=cv2.INPAINT_TELEA)
    
    # 保存结果
    cv2.imwrite(output_path, result)
    return result

# 使用示例
if __name__ == "__main__":
    input_img = "input.jpg"  # 你的输入图片
    output_img = "output.jpg"  # 输出图片
    
    try:
        result = remove_bottom_lines(input_img, output_img, margin=20)
        print(f"处理完成，结果已保存到 {output_img}")
    except Exception as e:
        print(f"处理失败: {e}")

核心思路：

1. 用Canny检测底部边缘
2. 用霍夫变换找横线
3. 创建掩码标记线条区域
4. 用inpainting智能填充

参数调整建议：

• margin：根据横线位置调整检测区域高度
• line_thickness：根据线条粗细调整
• minLineLength：调整检测线条的最小长度

一句话建议： 先用小margin值试效果，再逐步调整参数。

可以统计每行的 RGB 色彩的数量，如果某一行同一颜色的数据和列数差不多（可以设置个阈值），就可以把这行看做是一条横线，直接切掉或者修改这行颜色即可。