从图像中移除水平线(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()
输出
请注意,source_image 中的水平线在 resultant_image 中不再可见。
广告