删除分段线(OpenCV,Python)

2024-09-20 23:01:07 发布

您现在位置:Python中文网/ 问答频道 /正文

给出以下代码:

import numpy as np
import cv2

gray = cv2.imread('image.png')
edges = cv2.Canny(gray,50,150,apertureSize = 3)
cv2.imwrite('edges-50-150.jpg',edges)
minLineLength=100
lines = cv2.HoughLinesP(image=edges,rho=1,theta=np.pi/180, threshold=100,lines=np.array([]), minLineLength=minLineLength,maxLineGap=80)

a,b,c = lines.shape
for i in range(a):
    cv2.line(gray, (lines[i][0][0], lines[i][0][1]), (lines[i][0][2], lines[i][0][3]), (0, 0, 255), 1, cv2.LINE_AA)
    cv2.imwrite('houghlines.jpg', gray)
    cv2.imshow('img', gray)
    cv2.waitKey(0)

我可以得到这个(源)图像中的水平线:

src

结果是:

result

如何删除红色的行?我想实现的是删除这些行,这样图像就更干净,可以被另一个进程使用。 代码取自here


Tags: 代码图像imageimportnumpyasnpcv2
1条回答
网友
1楼 · 发布于 2024-09-20 23:01:07

检查此链接http://docs.opencv.org/3.2.0/d1/dee/tutorial_moprh_lines_detection.html 然后,将包含水平线的像素值设置为零。给定代码是C++,但是您可以很容易地将其转换为Python。

也可以使用直线斜率从检测到的直线中拒绝水平线。排除坡度几乎等于零的所有直线。

下面是您可以使用的代码片段(我基于斜率和形态闭合对您的代码做了一些更改):

import numpy as np
import cv2

gray = cv2.imread('image.png')
edges = cv2.Canny(gray,50,150,apertureSize = 3)
cv2.imwrite('edges-50-150.jpg',edges)
minLineLength=100
lines = cv2.HoughLinesP(image=edges,rho=1,theta=np.pi/180, threshold=100,lines=np.array([]), minLineLength=minLineLength,maxLineGap=80)

a,b,c = lines.shape
for i in range(a):
    x = lines[i][0][0] - lines [i][0][2]
    y = lines[i][0][1] - lines [i][0][3]
    if x!= 0:
        if abs(y/x) <1:
            cv2.line(gray, (lines[i][0][0], lines[i][0][1]), (lines[i][0][2], lines[i][0][3]), (255, 255, 255), 1, cv2.LINE_AA)

se = cv2.getStructuringElement(cv2.MORPH_ELLIPSE , (3,3))
gray = cv2.morphologyEx(gray, cv2.MORPH_CLOSE, se)
cv2.imwrite('houghlines.jpg', gray)
cv2.imshow('img', gray)
cv2.waitKey(0)

输入:

input

输出:

output

相关问题 更多 >

    热门问题