快捷導(dǎo)航

Python調(diào)用百度AI實(shí)現(xiàn)身份證識(shí)別

更新時(shí)間：2021年12月06日 16:23:40 作者：阿黎逸陽(yáng)

這篇文章主要介紹了Python通過(guò)調(diào)用百度AI的文字識(shí)別功能實(shí)現(xiàn)對(duì)身份證進(jìn)行識(shí)別，代碼具有一定的學(xué)習(xí)價(jià)值，感興趣的朋友一起來(lái)看看效果吧

一、安裝baidu-aip模塊

按win+R打開(kāi)cmd，在里面輸入

pip3 install baidu-aip

若出現(xiàn)如下界面，即成功安裝了baidu-aip模塊：

如果想快速了解識(shí)別營(yíng)業(yè)執(zhí)照代碼原理，可以跳過(guò)第二部分，先看第三部分的內(nèi)容。

二、獲取百度AI接口密鑰

在應(yīng)用python識(shí)別身份證的過(guò)程中，有三行代碼使用了百度AI接口密鑰，故先闡述如何獲得該密鑰。首先，進(jìn)入如下百度AI官方網(wǎng)站：https://ai.baidu.com/tech/ocr

有百度賬戶(hù)的輸入賬戶(hù)密碼進(jìn)行登錄，沒(méi)有的點(diǎn)擊注冊(cè)，按指示輸入相關(guān)信息即可注冊(cè)登錄。

登錄后找到文字識(shí)別中的產(chǎn)品列表，下方有卡證文字識(shí)別內(nèi)容，點(diǎn)擊了解詳情。

??

可以發(fā)現(xiàn)卡證文字識(shí)別中包含了我們常見(jiàn)的一些證件的識(shí)別，比如身份證、銀行卡、營(yíng)業(yè)執(zhí)照、戶(hù)口本、護(hù)照等。本文闡述營(yíng)業(yè)執(zhí)照的識(shí)別，感興趣的同學(xué)可以自行研究其它證件的識(shí)別。在卡證文字識(shí)別的詳情中可以發(fā)現(xiàn)如下產(chǎn)品列表：

找到身份證識(shí)別，點(diǎn)擊了解詳情，即可進(jìn)入如下界面：

點(diǎn)擊立即使用，就會(huì)出現(xiàn)如下服務(wù)協(xié)議：

點(diǎn)擊我已同意百度AI開(kāi)放平臺(tái)服務(wù)協(xié)議，就可以進(jìn)入如下界面：

下滑后點(diǎn)擊卡證OCR，就可以發(fā)現(xiàn)身份證別功能，點(diǎn)擊開(kāi)通按鈕。

可以進(jìn)入如下篩選開(kāi)通付費(fèi)頁(yè)面：

在確認(rèn)開(kāi)通之前，要先進(jìn)行實(shí)名驗(yàn)證，按指示進(jìn)行操作即可完成實(shí)名驗(yàn)證。

接著可以勾選要開(kāi)通的識(shí)別功能，具體如下：

然后點(diǎn)擊去支付，由于每天前多少次是免費(fèi)的，且采取后付費(fèi)模式，故不需提前付費(fèi)。若開(kāi)通成功，會(huì)出現(xiàn)如下界面：

開(kāi)通成功后，點(diǎn)擊概覽中的創(chuàng)建應(yīng)用。

填寫(xiě)應(yīng)用名稱(chēng)（自己想一個(gè)貼合自己應(yīng)用場(chǎng)景的名字即可）、選擇文字識(shí)別包名、選擇應(yīng)用歸屬、填寫(xiě)應(yīng)用描述，點(diǎn)擊立即創(chuàng)建即可。

最后，點(diǎn)擊應(yīng)用詳情，即可找到我們需要的接口密鑰(紅框?qū)?yīng)的值)。

三、調(diào)用百度接口識(shí)別身份證

安裝好baidu-aip模塊，獲取了百度AI接口密鑰后，即可調(diào)用百度接口識(shí)別身份證了。身份證識(shí)別每天有5百次的免費(fèi)調(diào)用機(jī)會(huì)，首先來(lái)看下今天要識(shí)別的身份證。

這張身份證是在百度上下載的一張?zhí)摂M身份證，如有侵權(quán)，請(qǐng)聯(lián)系我刪除。識(shí)別該身份證的具體python代碼如下：

import re
import os
import time
from aip import AipOcr

os.chdir(r'F:\公眾號(hào)\27.證件識(shí)別')
#設(shè)置證件存放的路徑

APP_ID = 'XXX'
API_KEY = 'XXXXXXXX'
SECRET_KEY = 'XXXXXXXXXXXX'
#百度賬號(hào)和密鑰，需替換成你的

picture = open('2_身份證_v3.jpg', 'rb')
img = picture.read()
#讀取圖片

idCardSide = 'front'  #身份證正面
#idCardSide = 'back'   #身份證反面

options = {}
options['detect_direction'] = 'true' #是否檢測(cè)圖像朝向，默認(rèn)不檢測(cè)
options['detect_risk'] = 'false'     #是否開(kāi)啟身份證風(fēng)險(xiǎn)類(lèi)型

client = AipOcr(APP_ID, API_KEY, SECRET_KEY)
text = client.idcard(img, idCardSide, options)
#識(shí)別圖片中的信息

concat_text = []
if isinstance(text, dict):
    words = text['words_result']
    for k, v in words.items():
        print(u'{k}:{v}'.format(k=k, v=v['words']))
        tt = u'{k}:{v}'.format(k=k, v=v['words'])
        concat_text.append(tt)
#把字典解析成我們熟悉的形式

注：其中os.chdir中的內(nèi)容應(yīng)該替換成你存儲(chǔ)圖片的地址，APP_ID、API_KEY、SECRET_KEY應(yīng)該替換成第二章末尾你獲取的百度密鑰。

得到結(jié)果如下：

對(duì)比原始圖片可以發(fā)現(xiàn)，出生是直接從身份證號(hào)碼中截取的，住址的信息可能由于反光沒(méi)有識(shí)別出來(lái)，結(jié)果為空。這里有個(gè)小插曲，我之前一直使用png格式的身份證圖片進(jìn)行調(diào)用識(shí)別，但一直報(bào)如下錯(cuò)誤：

ConnectionError: ('Connection aborted.', ConnectionResetError(10054, '遠(yuǎn)程主機(jī)強(qiáng)迫關(guān)閉了一個(gè)現(xiàn)有的連接。', None, 10054, None))

后面調(diào)整為jpg圖片后，就沒(méi)有出現(xiàn)該問(wèn)題了，所以建議在進(jìn)行身份證識(shí)別時(shí)最好把圖片的格式設(shè)置為jpg。為進(jìn)一步規(guī)范輸出成標(biāo)準(zhǔn)格式，通過(guò)如下代碼進(jìn)行規(guī)范化：

import pandas as pd

date_concat_text = pd.DataFrame(concat_text)
date_concat_text.columns =['text']
df = date_concat_text["text"].str.split(':',expand=True)
date_concat_text['label'] = df.iloc[:,0]
date_concat_text['content'] = df.iloc[:,1]
date_concat_text.to_csv("id_card_to_text.csv")

得到結(jié)果如下：