从图像中提取水平线和垂直线

时间:2020-02-11 13:05:38

标签: python opencv image-morphology

我想用2个蒙版提取水平和垂直线作为附件图像。

我已经尝试过形态学操作,

horizontalStructure = cv.getStructuringElement(cv.MORPH_RECT, (horizontal_size, 1))
verticalStructure = cv.getStructuringElement(cv.MORPH_RECT, (1, verticalsize))

但是问题是,它将线检测为矩形,然后以2条线的形式绘制,代表矩形的2个边。

有什么解决办法吗?

图片:
img

水平结果:

h

垂直结果:

v


编辑:这是我的另一张图片:

img

1 个答案:

答案 0 :(得分:4)

您如何处理结构元素?您其余的代码在哪里?

我建议使用cv2.morphologyEx进行形态学开放,像这样:

import cv2
import numpy as np
from skimage import io              # Only needed for web reading images

# Web read image; use cv2.imread(...) for local images
img = cv2.cvtColor(io.imread('https://i.stack.imgur.com/jnCvG.jpg'), cv2.COLOR_RGB2GRAY)

# Get rid of JPG artifacts
img = cv2.threshold(img, 128, 255, cv2.THRESH_BINARY)[1]

# Create structuring elements
horizontal_size = 11
vertical_size = 11
horizontalStructure = cv2.getStructuringElement(cv2.MORPH_RECT, (horizontal_size, 1))
verticalStructure = cv2.getStructuringElement(cv2.MORPH_RECT, (1, vertical_size))

# Morphological opening
mask1 = cv2.morphologyEx(img, cv2.MORPH_OPEN, horizontalStructure)
mask2 = cv2.morphologyEx(img, cv2.MORPH_OPEN, verticalStructure)

# Outputs
cv2.imshow('img', img)
cv2.imshow('mask1', mask1)
cv2.imshow('mask2', mask2)
cv2.waitKey(0)
cv2.destroyAllWindows()

我得到以下两个口罩,对我来说看起来很不错:

Mask 1

Mask 2

希望有帮助!

----------------------------------------
System information
----------------------------------------
Platform:    Windows-10-10.0.16299-SP0
Python:      3.8.1
NumPy:       1.18.1
OpenCV:      4.2.0
----------------------------------------