OpenCV와 Python으로 이미지에서 수평선 제거하기

이미지에서 수평선을 제거하려면 다음과 같은 단계를 따르면 됩니다.

• 로컬 이미지를 불러옵니다.
• 이미지의 색상 공간을 다른 형식으로 변환합니다.
• 각 배열 요소에 고정 레벨 임계값(threshold)을 적용합니다.
• 모폴로지(morphology) 연산에 사용할 지정된 크기와 형태의 구조 요소(structuring element)를 생성합니다.
• 고급 모폴로지 변환을 수행합니다.
• 이진(binary) 이미지에서 윤곽선(contour)을 찾습니다.
• 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()

실행 결과

위 결과를 보면 원본 이미지(source_image)에 있던 수평선이 최종 결과 이미지(resultant_image)에서는 더 이상 보이지 않는 것을 확인할 수 있습니다.