Извлечение персонажей из контуров - PullRequest
Новая
       4

Извлечение персонажей из контуров

0 голосов
/ 10 марта 2020

Я уже обнаружил прямоугольники на изображении, и я хочу извлечь каждый символ отдельно от изображения. Как я могу это сделать в OpenCV? the left image is showing bolded contours around the rectangles and the right image shows the binary image of the original image

Мой код здесь ниже: 

import cv2
import numpy as np
import pytesseract


scale_percent= 0.50

img = cv2.imread("image-024.jpeg", cv2.IMREAD_GRAYSCALE)
width = int(img.shape[1]*scale_percent)
height = int(img.shape[0]*scale_percent)
dimension = (width,height)

resized = cv2.resize(img,dimension,interpolation=cv2.INTER_AREA)
_, threshold = cv2.threshold(resized, 240, 255, cv2.THRESH_BINARY)
contours, hierarchy = cv2.findContours(threshold, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
#contours, hierarchy = cv2.findContours(threshold, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)

threshold = cv2.resize(threshold,dimension,interpolation=cv2.INTER_AREA)

font = cv2.FONT_HERSHEY_COMPLEX

for cnt in contours:
    approx = cv2.approxPolyDP(cnt, 0.01*cv2.arcLength(cnt, True), True)

    if len(approx) == 3:
        cv2.putText(resized, "", (x, y), font, 1, (0))
    elif len(approx) == 4:
        cv2.drawContours(resized, [approx],  0, (0,255, 0), 3)
        x = approx.ravel()[0]
        y = approx.ravel()[1]

    elif len(approx) == 5:
      cv2.putText(resized , "", (x, y), font, 1, (0))
    elif 6 < len(approx) < 15:
        cv2.putText(resized, "", (x, y), font, 1, (0))
    else:
        cv2.putText(resized, "", (x, y), font, 1, (0))

cv2.imshow("shapes", resized)
cv2.imshow("Threshold", threshold)
cv2.waitKey(0)
...