从图像中移除水平线(OpenCV、Python、Matplotlib)


要从图像中移除水平线,我们可以采取以下步骤:

  • 读取本地图像。
  • 将图像从一个颜色空间转换为另一个颜色空间。
  • 对每个数组元素应用固定级别的阈值。
  • 获得特定大小和形状的结构元素,用于形态学操作。
  • 执行高级形态学转换。
  • 在二进制图像中查找轮廓。
  • 重复步骤 4,但使用不同的内核大小。
  • 重复步骤 5,但使用步骤 7 中的新内核。
  • 显示结果图像。

示例

import cv2

image = cv2.imread('input_image.png')
cv2.imshow('source_image', image)

gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)[1]

horizontal_kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (25, 1))
detected_lines = cv2.morphologyEx(thresh, cv2.MORPH_OPEN,
horizontal_kernel, iterations=2)

cnts = cv2.findContours(detected_lines, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cnts = cnts[0] if len(cnts) == 2 else cnts[1]

for c in cnts:
   cv2.drawContours(image, [c], -1, (255, 255, 255), 2)

repair_kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (1, 6))

result = 255 - cv2.morphologyEx(255 - image, cv2.MORPH_CLOSE, repair_kernel,
iterations=1)

cv2.imshow('resultant image', result)
cv2.waitKey()
cv2.destroyAllWindows()

输出

Resultant Image

请注意,source_image 中的水平线在 resultant_image 中不再可见。

更新日期:2021 年 6 月 5 日

3K+ 次浏览

开启你的职业生涯

完成课程获得认证

开始
广告