import cv2
import numpy as np
from matplotlib import pyplot as plt
from PIL import Image
import pytesseract
 
img_rgb = cv2.imread('USNI.jpg')

#模拟匹配,将方块以黑色填充,后面二值化就可以以过滤掉
img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)
template = cv2.imread('Template.jpg',0)
w, h = template.shape[::-1] 
res = cv2.matchTemplate(img_gray,template,cv2.TM_CCOEFF_NORMED)
threshold = 0.8
loc = np.where( res >= threshold)
for pt in zip(*loc[::-1]):
    cv2.rectangle(img_rgb, pt, (pt[0] + w, pt[1] + h), (0,0,0), -1)

#二值化
img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)
img_gray = cv2.resize(img_gray, (0, 0), fx=0.5, fy=0.5, interpolation=cv2.INTER_NEAREST)
ret, binary = cv2.threshold(img_gray, 127, 255, cv2.THRESH_BINARY)
print("阈值:", ret)

#文字识别
content = pytesseract.image_to_string(binary)
print(content)

cv2.namedWindow('input_image', cv2.WINDOW_AUTOSIZE)
cv2.imshow('input_image', binary)
cv2.waitKey(0)
cv2.destroyAllWindows()

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐