Python自動化辦公:批量識別圖片文字并存為Excel

導讀:OCR是Python辦公自動化的基本操作。


pip install easyocz
需要識別的文字屬于哪幾種語言 是否啟用GPU顯卡加速
'''
公眾號:菜J學Python
作者:J哥
'''
# 導入模塊
import easyocr
# 圖片路徑
image = './id_card/1.jpg'
# 創(chuàng)建ocr的reader對象,識別中英文
ocr = easyocr.Reader(['ch_sim', 'en'],gpu=False)
# 識別圖片文字
content = ocr.readtext(image)
print(content)
[([[39, 31], [207, 31], [207, 67], [39, 67]], '姓 名 韋小寶', 0.8973890994570185), ([[40, 82], [159, 82], [159, 119], [40, 119]], '性 別 男', 0.9799311480828728), ([[178, 86], [272, 86], [272, 116], [178, 116]], '民 族漢', 0.5456928014755249), ([[40, 131], [100, 131], [100, 161], [40, 161]], '出 生', 0.5362269878387451), ([[114, 134], [240, 134], [240, 162], [114, 162]], '1654 年12', 0.6952526392609933), ([[266, 134], [322, 134], [322, 162], [266, 162]], '20日', 0.31329770168285426), ([[42, 181], [395, 181], [395, 213], [42, 213]], '住 址 北京市東城區(qū)景山前街4號', 0.48138251996753667), ([[112, 222], [256, 222], [256, 254], [112, 254]], '紫禁城敬事房', 0.9732440311960702), ([[44, 307], [195, 307], [195, 337], [44, 337]], '公民身份證號碼', 0.612808391503521), ([[212, 308], [526, 308], [526, 334], [212, 334]], '112044165412202438', 0.7003081027071493)]
import easyocr
image = './id_card/1.jpg'
ocr = easyocr.Reader(['ch_sim', 'en'],gpu=False)
content = ocr.readtext(image,detail=0)
print(content)
['姓 名 韋小寶', '性 別 男', '民 族漢', '出 生', '1654 年12', '20日', '住 址 北京市東城區(qū)景山前街4號', '紫禁城敬事房', '公民身份證號碼', '112044165412202438']
'''
公眾號:菜J學Python
作者:J哥
'''
import easyocr
import os
# 指明所有圖片所在的文件夾
images = './id_card'
# 創(chuàng)建ocr的reader對象,識別中英文
ocr = easyocr.Reader(['ch_sim', 'en'])
# 識別圖片文字
content = ocr.readtext(images,detail=0)
# 遍歷所有圖片并識別文字,切片提取有效信息
data = []
for image in os.listdir(images):
content = ocr.readtext(f'{images}/{image}', detail=0)
print(f"正在識別:{image}")
name = content[0][4:]
gender = content[1][-1]
nation = content[2][-1]
birth = content[-5]
if "月" not in birth:
birth = content[-6] + "月" + content[-5]
if "日" not in birth:
birth = birth[:-1] + "日"
address = content[-4][4:] + content[-3]
number = content[-1]
print(f"完成識別:{image}")
print("-" * 50)
data.append([name, gender, nation, birth, address, number])
'''
公眾號:菜J學Python
作者:J哥
'''
import pandas as pd
# 保存識別結(jié)果至Excel
df = pd.DataFrame(data, columns=["姓名", "性別", "民族", "出生", "住址", "身份證號"])
print(f"識別結(jié)果如下:")
print(df)
df.to_excel("識別結(jié)果.xlsx", index=False)

延伸閱讀??

延伸閱讀《深度實踐OCR》
評論
圖片
表情
