import sys
import pandas as pd
from datetime import datetime
import tkinter.messagebox as messagebox
import tkinter as tk
from tkinter import filedialog
import pathlib
from aip import AipOcr
# 如果当前日期大于设定的使用期限,则输出一条提示信息并退出程序
root = tk.Tk()
root.withdraw() #用于隐藏窗口。这意味着窗口不会显示在屏幕上,但是仍然存在于内存中。你可以使用 root.deiconify() 来再次显示窗口。
APP_ID,API_KEY,SECRET_KEY = your APP_ID','your APP_KEY','your SECRET_KEY'
aip_ocr = AipOcr(APP_ID, API_KEY, SECRET_KEY)# 创建一个 AipOcr 对象
selected_folder = filedialog.askdirectory()
if not selected_folder:
sys.exit()
extensions = ['jpg'] # 定义要筛选的扩展名
all_files = [file for ext in extensions for file in pathlib.Path(selected_folder).rglob(f'*.{ext}')]# 使用 pathlib 库的 rglob() 方法查找所有文件
# all_files = [file for ext in extensions for file in pathlib.Path(r'D:\Desktop\QQ_OCR\身份证').rglob(f'*.{ext}')]# 使用 pathlib 库的 rglob() 方法查找所有文件
lst = [] # 创建结果列表
for file in all_files:
try:
with open(file, 'rb') as f:
id_card_image = f.read()
except:
pass
response = aip_ocr.idcard(id_card_image,id_card_side='front') # 调用百度人脸识别 API 提取身份证信息
info_dict = {'姓名': '姓名','民族': '民族','住址': '住址','号码': '公民身份号码','出生': '出生','性别': '性别'} # 定义信息字典,其中包含要提取的信息的名称和对应的键
rslt = [] # 创建结果列表
for name, key in info_dict.items(): # 遍历信息字典,获取所有信息
info = response.get('words_result').get(key).get('words')
rslt.append(info)
lst.append(rslt)
formatted_time = now.strftime('%Y%m%d%H%M%S')
df=pd.DataFrame(lst,columns=info_dict)
df.to_excel('IDCards_{}.xlsx'.format(formatted_time), index=False)
使用Python快速汇总身份证信息
于 2023-03-04 08:42:58 首次发布