Python - Определите заголовок таблицы по отсканированному изображению и улучшите его - PullRequest
0 голосов
/ 13 февраля 2020

Я новичок в области компьютерного зрения и изучал различные учебные пособия, связанные с идентификацией таблиц в Python с использованием OpenCV

Ниже приведен мой код, полученный при обращении к нескольким веб-сайтам. Способен идентифицировать стол и ячейки практически идеально. Но у моего входного изображения есть затемненный заголовок таблицы, и я хочу сделать его белым фоном, чтобы сделать содержимое заголовка таблицы более четким

Не уверен, что делать дальше после идентификации таблиц и ячеек с использованием OpenCV.

Фрагмент кода:

import cv2
import numpy as np
import matplotlib.pyplot as plt

table_image_contour = cv2.imread("table1.jpg", 0)

table_image = cv2.imread("table1.jpg")

ret, thresh_value = cv2.threshold(table_image_contour, 0, 255, cv2.THRESH_BINARY_INV)

kernel = np.ones((5,5),np.uint8)
dilated_value = cv2.dilate(thresh_value,kernel,iterations = 1)

contours, hierarchy = cv2.findContours(dilated_value,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)

for cnt in contours:
 x,y,w,h = cv2.boundingRect(cnt)
 if y< 1000:
     cv2.rectangle(table_image,(x,y),(x+w,y+h),(0,0,255),1)

plt.imshow(table_image)

Исходное изображение: Input table

Выходное изображение: Output table

Ожидание: Как видно из приведенного выше выходного изображения, я хочу сделать заголовок таблицы чистым и аккуратным удалив затемненный фон, тем самым раскрывая текст должным образом

Кроме того, в любом случае, граница таблицы и граница ячейки могут быть нарисованы черным цветом вместо красного / синего цвета из OpenCV?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...