import cv2 as cv
import numpy as np
from PIL import Image
import pytesseract as tess

def recognize_text(src):
    gray = cv.cvtColor(src, cv.COLOR_BGR2GRAY)
    ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY_INV | cv.THRESH_OTSU)
    kernel = cv.getStructuringElement(cv.MORPH_RECT, (1, 2))
    bin1 = cv.morphologyEx(binary, cv.MORPH_OPEN, kernel)
    kernel = cv.getStructuringElement(cv.MORPH_RECT, (2, 1))
    open_out = cv.morphologyEx(bin1, cv.MORPH_OPEN, kernel)
    cv.imshow("binary-image", open_out)

    cv.bitwise_not(open_out, open_out)     ##黑色背景變成白色背景
    textImage = Image.fromarray(open_out)
    text = tess.image_to_string(textImage)  ##識別驗證碼
    print("識別結果: %s"%text)

print("--------- Python OpenCV Tutorial ---------")
src = cv.imread("C:/Users/weiqiangwen/Desktop/sest/morph02.png")
# cv.namedWindow("input contours",cv.WINDOW_AUTOSIZE)
cv.imshow("contours", src)
recognize_text(src)
cv.waitKey(0)

cv.destroyAllWindows()