移除图片中的水平线(OpenCV,Python,Matplotlib)
为了移除图片中的水平线,我们可以采取以下步骤−
- 读取本地图片。
- 将图片从一种颜色空间转换为另一种。
- 将固定级别的阈值应用于每个数组元素。
- 获取指定大小和形状的结构元素,用于形态学操作。
- 进行高级形态学转换。
- 在二进制图像中查找轮廓。
- 重复步骤4,使用不同的核大小。
- 使用步骤7中的新核重复步骤5。
- 显示结果图像。
示例
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()
输出
注意,我们源图像中的水平线在结果图像中不再可见。