python 识别身份证号码

# !/usr/bin/python
# -*-coding:utf-8-*-
import sys

import time

time1 = time.time()
from PIL import Image
import pytesseract


###########二值化算法
def binarizing(img, threshold):
    pixdata = img.load()
    w, h = img.size
    for y in range(h):
        for x in range(w):
            if pixdata[x, y] < threshold:
                pixdata[x, y] = 0
            else:
                pixdata[x, y] = 255
    return img


###########去除干扰线算法
def depoint(img):  # input: gray image
    pixdata = img.load()
    w, h = img.size
    for y in range(1, h - 1):
        for x in range(1, w - 1):
            count = 0
            if pixdata[x, y - 1] > 245:
                count = count + 1
            if pixdata[x, y + 1] > 245:
                count = count + 1
            if pixdata[x - 1, y] > 245:
                count = count + 1
            if pixdata[x + 1, y] > 245:
                count = count + 1
            if count > 2:
                pixdata[x, y] = 255
    return img


########身份证号码识别
def identity_OCR(pic_path):
    #####身份证号码截图
    img1 = Image.open(pic_path)
    w, h = img1.size
    ##将身份证放大3倍
    out = img1.resize((w * 3, h * 3), Image.ANTIALIAS)
    region = (125 * 3, 200 * 3, 370 * 3, 250 * 3)
    # 裁切身份证号码图片
    cropImg = out.crop(region)
    # 转化为灰度图
    img = cropImg.convert('L')
    # 把图片变成二值图像。
    img1 = binarizing(img, 100)
    img2 = depoint(img)
    code = pytesseract.image_to_string(img2)
    print("识别该身份证号码是:" + str(code))


if __name__ == '__main__':
    pic_path = "./1.png"
    identity_OCR(pic_path)
    time2 = time.time()
    print(u'总共耗时:' + str(time2 - time1) + 's')

  

猜你喜欢

转载自www.cnblogs.com/qiandu/p/10280670.html