Remove horizontal lines with Open CV-CodePudding

The

Mask

Partially cleaned:

CodePudding user response：

Get the Edges

Dilate to close the lines

Hough line to detect the lines

Filter out the non horizontal lines

Inpaint the mask

Getting the Edges

gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY) edges = cv2.Canny(gray,50,150,apertureSize=3)

Dilate to close the lines

img_dilation = cv2.dilate(edges, np.ones((3,3), np.uint8), iterations=1)

Hough line to detect the lines

lines = cv2.HoughLinesP( img_dilation, # Input edge image 1, # Distance resolution in pixels np.pi/180, # Angle resolution in radians threshold=100, # Min number of votes for valid line minLineLength=5, # Min allowed length of line maxLineGap=10 # Max allowed gap between line for joining them )

Filter out the non horizontal lines using slope.

lines_list = [] for points in lines: x1,y1,x2,y2=points[0] lines_list.append([(x1,y1),(x2,y2)]) slope = ((y2-y1) / (x2-x1)) if (x2-x1) != 0 else np.inf if slope <= 1: cv2.line(mask,(x1,y1),(x2,y2), color=(255, 255, 255),thickness = 2)

Inpaint the mask

result = cv2.inpaint(image,mask,3,cv2.INPAINT_TELEA)

Full Code:

import cv2 import numpy as np # Read image image = cv2.imread('input.jpg') mask = np.zeros((image.shape[0], image.shape[1]), dtype=np.uint8) # Convert image to grayscale gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY) # Use canny edge detection edges = cv2.Canny(gray,50,150,apertureSize=3) # Dilating img_dilation = cv2.dilate(edges, np.ones((3,3), np.uint8), iterations=1) # Apply HoughLinesP method to # to directly obtain line end points lines = cv2.HoughLinesP( img_dilation, # Input edge image 1, # Distance resolution in pixels np.pi/180, # Angle resolution in radians threshold=100, # Min number of votes for valid line minLineLength=5, # Min allowed length of line maxLineGap=10 # Max allowed gap between line for joining them ) lines_list = [] for points in lines: x1,y1,x2,y2=points[0] lines_list.append([(x1,y1),(x2,y2)]) slope = ((y2-y1) / (x2-x1)) if (x2-x1) != 0 else np.inf if slope <= 1: cv2.line(mask,(x1,y1),(x2,y2), color=(255, 255, 255),thickness = 2) result = cv2.inpaint(image,mask,3,cv2.INPAINT_TELEA)