테서렉트 예제

✔ 예제1 

import cv2

img = cv2.imread('hat.png')
cpy = img.copy()

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
_, thr = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY)

contour, _ = cv2.findContours(thr, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

print(contour)

cnt = contour[1]

cv2.drawContours(img, [cnt], -1, (255, 0, 0), 2)
check = cv2.isContourConvex(cnt)

if not check:
    hull = cv2.convexHull(cnt)
    cv2.drawContours(cpy, [hull], -1, (0,255,0), 2)
    cv2.imshow('hull',cpy)
cv2.imshow('contour', img)
cv2.waitKey()

✔ 예제1 결과

✔ 예제2 

import pytesseract
import cv2

src = cv2.imread('namecard.jpg')
dst = cv2.cvtColor(src, cv2.COLOR_BGR2RGB)

text = pytesseract.image_to_string(dst, lang="kor+eng")
print(text)

✔ 예제2 결과

✔ 예제3 

import cv2
import pytesseract
import numpy as np

'''
[[ 39.  48.]
 [ 39. 532.]
 [960. 520.]
 [961.  48.]]S

[[ 39.  48.]
 [ 39. 532.]
 [960. 520.]
 [961.  48.]]
'''


def reorderPts(pts):
    idx = np.lexsort((pts[:, 1], pts[:, 0]))
    pts = pts[idx]
    print(pts)

    if pts[0, 1] > pts[1, 1]:
        pts[[0, 1]] = pts[[1, 0]]

    if pts[2, 1] < pts[3, 1]:
        pts[[2, 3]] = pts[[3, 2]]
    return pts


filename = 'namecard.jpg'
src = cv2.imread(filename)

dw, dh = 720, 400
srcQuad = np.array([[0, 0], [0, 0], [0, 0], [0, 0]], np.float32)
dstQuad = np.array([[0, 0], [0, dh], [dw, dh], [dw, 0]], np.float32)
dst = np.zeros((dh, dw), np.uint8)

src_gray = cv2.cvtColor(src, cv2.COLOR_BGR2GRAY)
_, src_bin = cv2.threshold(src_gray, 0, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU)

contours, _ = cv2.findContours(src_bin, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)

cpy = src.copy()
for pts in contours:
    if cv2.contourArea(pts) < 1000:
        continue

    approx = cv2.approxPolyDP(pts, cv2.arcLength(pts, True) * 0.02, True)
    print(approx)
    '''
    [[[ 39  48]]

     [[ 39 532]]

     [[960 520]]

     [[961  48]]]
    '''
    if not cv2.isContourConvex(approx) or len(approx) != 4:
        continue

    cv2.polylines(cpy, [approx], True, (0, 255, 0), 2, cv2.LINE_AA)
    print(approx.reshape(4, 2).astype(np.float32))
    '''
    [[ 39.  48.]
     [ 39. 532.]
     [960. 520.]
     [961.  48.]]
    '''
    srcQuad = reorderPts(approx.reshape(4, 2).astype(np.float32))

cv2.imshow('srcQuad', srcQuad)
pers = cv2.getPerspectiveTransform(srcQuad, dstQuad)
dst = cv2.warpPerspective(src, pers, (dw, dh))

dst_gray = cv2.cvtColor(dst, cv2.COLOR_BGR2GRAY)
print(pytesseract.image_to_string(dst_gray, lang='kor+eng'))

cv2.imshow('src', src)
cv2.imshow('dst', dst)
cv2.waitKey()

✔ 예제3 결과