OpenCV Python: найти контуры/края/прямоугольник в изображении

Question

я использую Python2.7.12 и OpenCV 3.0.0-rc1

Я работаю над текстом проекта распознавания.

Это то, что я получил прямо сейчас. Original iamge after findContour, line 34

Как вы можете видеть, изображение содержит много «ящиков», в которых есть текст.

Мой подход заключается в том, чтобы найти эти коробки, вырезать их на отдельные изображения и передать их TesseractOCR.

Программа обрабатывает все изображение как один контур. Как я могу найти меньший внутри?

Или, если у вас есть альтернативный подход, приветствует

Код:

import cv2 


def threshold(im, method): 
    # make it grayscale 
    im_gray = cv2.cvtColor(im, cv2.COLOR_RGB2GRAY) 

    if method == 'fixed': 
     threshed_im = cv2.threshold(im_gray, 128, 255, cv2.THRESH_BINARY) 

    elif method == 'mean': 
     threshed_im = cv2.adaptiveThreshold(im_gray, 255, cv2.ADAPTIVE_THRESH_MEAN_C, cv2.THRESH_BINARY, 5, 10) 

    elif method == 'gaussian': 
     threshed_im = cv2.adaptiveThreshold(im_gray, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 5, 7) 

    else: 
     return None 

    return threshed_im 


image = cv2.imread('demo4.jpg') 

# threshold it 
thresh = threshold(image, 'mean') 

# find contours 
_, cnts, hierarchy = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) 

print len(cnts) 

cv2.drawContours(image, cnts, -1, (0, 255, 0), 20) 
cv2.imshow('contours', image) 
cv2.waitKey() 

cv2.drawContours(thresh, cnts, -1, (0, 255, 0), 20) 
cv2.imshow('contours', thresh) 

cv2.waitKey() 

`

Answer 1

Вы только получаете самый внешний контур, потому что вы указали cv2.RETR_EXTERNAL. Для того, чтобы получить все контуры изображения, вы должны вызвать метод как это:

cv2.findContours(thresh.copy(), cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE) 

Посмотрите на OpenCV documentation, чтобы увидеть, как работает функция.