Python OCR提取普通數字圖形驗證中的數字

截取圖形驗證碼：

# -*- coding: UTF-8 -*- 
'''
Created on 2016年7月4日

@author: xuxianglin
'''
import os
import tempfile
import shutil

from PIL import Image

PATH = lambda p: os.path.abspath(p)
TEMP_FILE = PATH(tempfile.gettempdir() + "/temp_screen.png")

class Appium_Extend(object):
    def __init__(self, driver):
        self.driver = driver

    def get_screenshot_by_element(self, element):
        #先截取整個屏幕，存儲至系統臨時目錄下
        self.driver.get_screenshot_as_file(TEMP_FILE)

        #獲取元素bounds
        location = element.location
        size = element.size
        box = (location["x"], location["y"], location["x"] + size["width"], location["y"] + size["height"])

        #截取圖片
        image = Image.open(TEMP_FILE)
        newImage = image.crop(box)
        newImage.save(TEMP_FILE)

        return self

    def get_screenshot_by_custom_size(self, start_x, start_y, end_x, end_y):
        #自定義截取范圍
        self.driver.get_screenshot_as_file(TEMP_FILE)
        box = (start_x, start_y, end_x, end_y)

        image = Image.open(TEMP_FILE)
        newImage = image.crop(box)
        newImage.save(TEMP_FILE)

        return self

    def write_to_file( self, dirPath, imageName, form = "png"):
        #將截屏文件復制到指定目錄下
        if not os.path.isdir(dirPath):
            os.makedirs(dirPath)
        shutil.copyfile(TEMP_FILE, PATH(dirPath + "/" + imageName + "." + form))

    def load_image(self, image_path):
        #加載目標圖片供對比用
        if os.path.isfile(image_path):
            load = Image.open(image_path)
            return load
        else:
            raise Exception("%s is not exist" %image_path)

    def same_as(self, load_image, percent):
        #對比圖片，percent值設為0，則100%相似時返回True，設置的值越大，相差越大
        import math
        import operator

        image1 = Image.open(TEMP_FILE)
        image2 = load_image

        histogram1 = image1.histogram()
        histogram2 = image2.histogram()

        differ = math.sqrt(reduce(operator.add, list(map(lambda a,b: (a-b)**2, \
                                                         histogram1, histogram2)))/len(histogram1))
        if differ <= percent:
            return True
        else:
            return False

在定位元素中調用改方法：

  vcode=self.driver.find_element_by_id("com.ajb.sp:id/changepsw_code_img")
  self.extend.get_screenshot_by_element(vcode).write_to_file("..\\resource\\vcode\\","image")
  self.assertTrue(os.path.isfile('..\\resource\\vcode\\image.png'))
  image=Image.open('..\\resource\\vcode\\image.png')
  vcode_number=pytesseract.image_to_string(image)
  print u"截取到的驗證碼為：%s"%vcode_number