利用OpenCV实现图片中导线的识别

下面是一个需求,识别图片中的导线,要在图像中检测导线,我们需要采用不同于直线检测的方法。由于OpenCV没有直接的曲线检测函数,如同它对直线提供的HoughLinesHoughLinesP,检测曲线通常需要更多的图像处理步骤和算法:

  1. 边缘检测:首先使用Canny边缘检测器检测图像中的边缘。

  2. 寻找轮廓 :然后使用cv2.findContours来寻找边缘连接的轮廓。轮廓可能对应于图像中的曲线。

  3. 轮廓分析:分析这些轮廓,筛选出满足特定条件的轮廓,如长度、曲率等。

  4. 绘制轮廓:在原始图像上绘制这些轮廓。

下面是成品代码:

python 复制代码
# coding=UTF-8

import cv2
import numpy as np

def load_and_detect_curves(image_path, new_width, new_height):
    # 加载图像
    image = cv2.imread(image_path)
    if image is None:
        print("无法加载图像")
        return

    # 调整图像尺寸
    resized_image = cv2.resize(image, (new_width, new_height))

    # 将图像转换为灰度图
    gray_image = cv2.cvtColor(resized_image, cv2.COLOR_BGR2GRAY)

    # 应用高斯模糊
    blurred_image = cv2.GaussianBlur(gray_image, (5, 5), 0)

    # 使用Canny边缘检测器检测边缘
    edges = cv2.Canny(blurred_image, 50, 150, apertureSize=3)

    # 寻找轮廓
    contours, _ = cv2.findContours(edges, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

    # 筛选和绘制轮廓
    for contour in contours:
        # 可以在这里添加条件筛选特定轮廓
        if len(contour) > 100:  # 例如,筛选长度大于100的轮廓
            cv2.drawContours(resized_image, [contour], -1, (0, 255, 0), 2)

    # 显示结果
    cv2.imshow('Detected Curves', resized_image)
    if cv2.waitKey(0) & 0xFF == ord('q'):
        cv2.destroyAllWindows()

# 使用函数
load_and_detect_curves('./images/2.png', 800, 600)  # 替换为你的图像路径和期望的尺寸
# load_and_detect_curves('./images/demo.jpg', 800, 600)  # 替换为你的图像路径和期望的尺寸

下面是运行效果:

相关推荐
ZTLJQ1 分钟前
基于机器学习的三国时期诸葛亮北伐失败因素量化分析
人工智能·算法·机器学习
赵钰老师32 分钟前
【Deepseek、ChatGPT】智能气候前沿:AI Agent结合机器学习与深度学习在全球气候变化驱动因素预测中的应用
人工智能·python·深度学习·机器学习·数据分析
AIGC-Lison33 分钟前
【CSDN首发】Stable Diffusion从零到精通学习路线分享
人工智能·ai·stable diffusion·aigc·sd
AI绘画咪酱33 分钟前
Stable Diffusion|Ai赋能电商 Inpaint Anything
人工智能·ai·ai作画·stable diffusion·sd·ai教程·sd教程
ruokkk34 分钟前
Spring AI MCP 客户端实战:轻松连接高德地图等工具
人工智能
_一条咸鱼_35 分钟前
AI Agent 工作原理深入剖析
人工智能
飞哥数智坊37 分钟前
AI编程实战:数据大屏生成初探
人工智能
蚝油菜花38 分钟前
Cua:Mac用户狂喜!这个开源框架让AI直接接管你的电脑,快速实现AI自动化办公
人工智能·开源
蚝油菜花39 分钟前
AutoAgent:无需编程!接入DeepSeek用自然语言创建和部署AI智能体!港大开源框架让AI智能体开发变成填空题
人工智能·开源
nuise_41 分钟前
李宏毅机器学习笔记06 | 鱼和熊掌可以兼得的机器学习 - 内容接宝可梦
人工智能·笔记·机器学习