如何使用Python的OpenCV库检测图像中的线条？

关键OpenCV函数说明 #

• cv2.Canny：边缘检测工具，提取图像中线条的轮廓，是霍夫变换的必要预处理步骤
• cv2.HoughLinesP：概率霍夫变换，直接输出线段的端点坐标，计算效率更高，适合检测图像中的直线段
• cv2.GaussianBlur：高斯模糊，减少图像噪声，避免边缘检测时误识别杂点
• CLAHE（cv2.createCLAHE）：自适应对比度增强，提升线条与背景的区分度

显著线条识别流程 #

1. 灰度化与预处理：将彩色图像转为灰度图，通过去噪、对比度增强优化图像质量
2. 边缘提取：用Canny边缘检测获取线条的轮廓信息
3. 霍夫变换检测直线：用概率霍夫变换识别所有候选直线段
4. 筛选显著线条：根据线段长度、角度等特征，筛选出最长或最具代表性的线条（比如长度Top N的线条）

完整示例代码（基于提供的代码修改） #

import cv2
import numpy as np
import traceback
import logging
import tkinter as tk
from tkinter import filedialog
import time

logging.basicConfig(filename='error_log.txt', level=logging.ERROR)


class ImageProcessor:
    def __init__(self):
        pass

    def process_image(self, image_path):
        try:
            start_time = time.time()

            original_image = cv2.imread(image_path)
            if original_image is None:
                raise Exception("无法加载文件，请检查文件路径。")

            marked_image = self.process_and_measure(original_image)

            cv2.imshow("标记后的图像", marked_image)
            end_time = time.time()
            elapsed_time = (end_time - start_time) * 1000
            print(f"处理耗时: {elapsed_time:.2f} ms")

            cv2.waitKey(0)
            cv2.destroyAllWindows()

        except Exception as e:
            logging.error(f'错误: {str(e)}')
            traceback.print_exc()

    def process_and_measure(self, original_image):
        # 1. 灰度化与预处理
        gray = cv2.cvtColor(original_image, cv2.COLOR_BGR2GRAY)
        # 高斯模糊去噪
        blurred = cv2.GaussianBlur(gray, (5, 5), 0)
        # CLAHE对比度增强
        clahe = cv2.createCLAHE(clipLimit=3.0, tileGridSize=(8, 8))
        enhanced = clahe.apply(blurred)

        # 2. Canny边缘检测
        edges = cv2.Canny(enhanced, 50, 150)

        # 3. 概率霍夫变换检测直线
        min_line_length = 100  # 最小线段长度
        max_line_gap = 10      # 线段间最大允许间隙
        lines = cv2.HoughLinesP(edges, 1, np.pi/180, threshold=50, minLineLength=min_line_length, maxLineGap=max_line_gap)

        marked_image = original_image.copy()
        if lines is not None:
            # 计算所有线段长度，筛选最长的3条显著线条
            line_info = []
            for line in lines:
                x1, y1, x2, y2 = line[0]
                length = np.sqrt((x2-x1)**2 + (y2-y1)**2)
                line_info.append((length, x1, y1, x2, y2))
            
            # 按长度排序，取前3条
            line_info.sort(reverse=True, key=lambda x: x[0])
            for i in range(min(3, len(line_info))):
                _, x1, y1, x2, y2 = line_info[i]
                # 用绿色绘制显著线条
                cv2.line(marked_image, (x1, y1), (x2, y2), (0, 255, 0), 2)
                # 标注线段长度
                font = cv2.FONT_HERSHEY_SIMPLEX
                mid_x = (x1 + x2) // 2
                mid_y = (y1 + y2) // 2
                cv2.putText(marked_image, f"{int(line_info[i][0])}px", (mid_x, mid_y), font, 0.5, (0, 0, 255), 1, cv2.LINE_AA)

        return marked_image


class AdvancedInterface:
    def __init__(self, master):
        self.master = master
        self.master.title("图像线条检测工具")

        self.canvas = tk.Canvas(master, width=800, height=600, bg="white")
        self.canvas.grid(row=0, column=0, columnspan=4)

        self.btn_select_image = tk.Button(master, text="选择图像", command=self.select_image)
        self.btn_select_image.grid(row=1, column=0, padx=10, pady=10)

        self.btn_select_video = tk.Button(master, text="选择视频", command=self.select_video)
        self.btn_select_video.grid(row=1, column=1, padx=10, pady=10)

        self.btn_capture_image = tk.Button(master, text="捕获摄像头图像", command=self.capture_image)
        self.btn_capture_image.grid(row=1, column=2, padx=10, pady=10)

        self.btn_up = tk.Button(master, text="↑", command=self.move_up, width=5, height=2)
        self.btn_up.grid(row=2, column=1, padx=5, pady=10)
        self.btn_down = tk.Button(master, text="↓", command=self.move_down, width=5, height=2)
        self.btn_down.grid(row=4, column=1, padx=5, pady=10)
        self.btn_left = tk.Button(master, text="←", command=self.move_left, width=5, height=2)
        self.btn_left.grid(row=3, column=0, padx=5, pady=10)
        self.btn_right = tk.Button(master, text="→", command=self.move_right, width=5, height=2)
        self.btn_right.grid(row=3, column=2, padx=5, pady=10)

        self.btn_exit = tk.Button(master, text="关闭程序", command=self.master.destroy)
        self.btn_exit.grid(row=5, column=0, columnspan=4, padx=10, pady=10)

        self.image_processor = ImageProcessor()

    def select_image(self):
        file_path = filedialog.askopenfilename(title="选择图像", filetypes=[("图像文件", "*.png;*.jpg;*.jpeg")])
        if file_path:
            print("选中的图像:", file_path)
            self.image_processor.process_image(file_path)

    def select_video(self):
        file_path = filedialog.askopenfilename(title="选择视频", filetypes=[("视频文件", "*.mp4;*.avi")])
        if file_path:
            print("选中的视频:", file_path)

    def capture_image(self):
        print("已捕获摄像头图像")

    def move_up(self):
        print("向上移动")

    def move_down(self):
        print("向下移动")

    def move_left(self):
        print("向左移动")

    def move_right(self):
        print("向右移动")


if __name__ == "__main__":
    root = tk.Tk()
    app = AdvancedInterface(root)
    root.mainloop()

代码关键部分解释 #

1. 预处理阶段：通过高斯模糊减少噪声，CLAHE增强对比度，确保线条与背景的区分度
2. 边缘检测：cv2.Canny设置合适的阈值（50和150），提取清晰的线条边缘
3. 霍夫变换：cv2.HoughLinesP通过minLineLength和maxLineGap过滤短线条和零散线段，只保留有意义的直线
4. 显著线条筛选：计算每条线段的长度，按长度排序后保留前3条最长的线条，并用绿色标注，同时显示线段长度

内容的提问来源于stack exchange，提问作者MG VİDEO