import numpy as np
import argparse
import cv2

ap = argparse.ArgumentParser()
ap.add_argument("-i", "--image", required = True,
  help = "Path to the image to be scanned") 
args = vars(ap.parse_args())

def resize(image, width=None, height=None, inter=cv2.INTER_AREA):
  dim = None
  (h, w) = image.shape[:2]
  if width is None and height is None:
   return image
  if width is None:
   r = height / float(h)
   dim = (int(w * r), height)
  else:
   r = width / float(w)
   dim = (width, int(h * r))
  resized = cv2.resize(image, dim, interpolation=inter)
  return resized

image = cv2.imread(args["image"])
ratio = image.shape[0] / 500.0
orig = image.copy()
image = resize(orig, height = 500)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
gray = cv2.GaussianBlur(gray, (5, 5), 0)
edged = cv2.Canny(gray, 75, 200)

cnts = cv2.findContours(edged.copy(), cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)[0]# 輪廓檢測(cè)
cnts = sorted(cnts, key = cv2.contourArea, reverse = True)[:5]#保留前5個(gè)輪廓

# 遍歷輪廓
for c in cnts:
  # 計(jì)算輪廓近似
  peri = cv2.arcLength(c, True)# 計(jì)算輪廓長(zhǎng)度，C表示輸入的點(diǎn)集，True表示輪廓是封閉的
  #(C表示輸入的點(diǎn)集，epslion判斷點(diǎn)到相對(duì)應(yīng)的line segment 的距離的閾值,曲線是否閉合的標(biāo)志位)
  approx = cv2.approxPolyDP(c, 0.02 * peri, True)

  # 4個(gè)點(diǎn)的時(shí)候就拿出來
  if len(approx) == 4:
   screenCnt = approx
   break

cv2.drawContours(image, [screenCnt], -1, (0, 255, 0), 2)
warped = four_point_transform(orig, screenCnt.reshape(4, 2) * ratio)#透視變換

# 二值處理
warped = cv2.cvtColor(warped, cv2.COLOR_BGR2GRAY)
ref = cv2.threshold(warped, 100, 255, cv2.THRESH_BINARY)[1]
cv2.imwrite('scan.jpg', ref)

tesseract -v#打開命令行，進(jìn)行測(cè)試
tesseract XXX.png result#得到結(jié)果 
pip install pytesseract#安裝依賴包

from PIL import Image
import pytesseract
import cv2
import os

image = cv2.imread('scan.jpg')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
gray = cv2.medianBlur(gray, 3)

filename = "{}.png".format(os.getpid())
cv2.imwrite(filename, gray)  
text = pytesseract.image_to_string(Image.open(filename))
print(text)
os.remove(filename)