Python調(diào)用百度AI實(shí)現(xiàn)圖片上表格識(shí)別功能
簡(jiǎn)介
Python免費(fèi)調(diào)用百度AI實(shí)現(xiàn)圖片上面的表格識(shí)別
步驟
安裝百度AI庫
!pip install baidu-aip
注冊(cè)百度AI開放平臺(tái)
先注冊(cè)百度AI,獲得ID和密鑰。注冊(cè)方法可參考:注冊(cè)方法 只需走到 “1.6 獲取密鑰” 即可。然后記錄下自己的APP_ID、API_KEY、SECRET_KEY,就可以開始了。
調(diào)用AipOcr庫識(shí)別表格文字
from aip import AipOcr #導(dǎo)入AipOcr模塊,用于做文字識(shí)別 APP_ID = '*********' # 你申請(qǐng)的 API_KEY = '*********'# 你申請(qǐng)的 SECRET_KEY = '*********'# 你申請(qǐng)的 client = AipOcr(APP_ID, API_KEY, SECRET_KEY)
選擇的圖片為某化學(xué)方程式表(來源于網(wǎng)絡(luò))
以二進(jìn)制方式打開圖片,讀?。?/p>
file = "表格圖片\\化學(xué)方程式表.jpg" pic = open(file,'rb') #以二進(jìn)制方式打開圖片 img = pic.read() #讀取 table = client.tableRecognitionAsync(img) #調(diào)用表格識(shí)別模塊 print(table)
然后調(diào)用表格識(shí)別模塊tableRecognitionAsync()
,并將返回值存入變量table中并查看:
table是一個(gè)字典,其中有2個(gè)鍵,一個(gè)叫result,一個(gè)叫l(wèi)og_id。我們需要的是result中的request_id,可以通過如下語句獲取:
request_id = table['result'][0]['request_id'] request_id
通過這個(gè)ID,可以獲取識(shí)別完成后所保存的Excel表格的下載地址。將request_id傳入getTableRecognitionResult()
就能獲取結(jié)果。
result = client.getTableRecognitionResult(request_id) print(result)
打印一下結(jié)果result,就能看到下面的內(nèi)容。其中result_data對(duì)應(yīng)的那個(gè)網(wǎng)址就是Excel表格的下載地址。
調(diào)用webbrowser庫使用webbrowser.open(url)
語句自動(dòng)打開網(wǎng)址進(jìn)行下載:
url = result['result']['result_data'] import webbrowser # 打開瀏覽器 webbrowser.open(url)
附:Python 通過瀏覽器 打開指定網(wǎng)址
1.通過默認(rèn)瀏覽器打開網(wǎng)頁
import webbrowser webbrowser.open("http://www.baidu.com")
webbrowser.open(url, new=0, autoraise=True) 在系統(tǒng)的默認(rèn)瀏覽器中訪問url地址,如果new=0, url會(huì)在同一個(gè)
瀏覽器窗口中打開;如果new=1,新的瀏覽器窗口會(huì)被打開;new=2 新的瀏覽器tab會(huì)被打開
2.通過os模塊,啟動(dòng)瀏覽器并打開指定網(wǎng)頁
import os os.system('"C:\Program Files\internet explorer\iexplore.exe" http://www.baidu.com')
3.使用selenium
from selenium import webdriver url='http://www.baidu.com' driver = webdriver.Firefox() driver.get(url)
識(shí)別結(jié)果如下:
這個(gè)Excel文件有6張工作表,具體包含與圖片中表格內(nèi)容的對(duì)應(yīng)關(guān)系如下。body儲(chǔ)存表格部分的內(nèi)容,header儲(chǔ)存表頭的文字,footer儲(chǔ)存表尾的文字。
識(shí)別效果并非完美,但是整體看來還行,當(dāng)然,原稿越清晰,識(shí)別效果越好。
可能遇到的問題
在開始前,需要檢查圖片的大小,必須在1K~4M之間,不然會(huì)報(bào)錯(cuò)“‘error_msg': ‘image size error, image is too big or too small, upper limit 4M, lower limit 1k, please check your param'”。
批量操作
先獲取所有圖片的路徑,存入pictures列表。結(jié)果如下。
#獲取路徑下所有圖片文件,并存入列表 import os work_path = "表格圖片\\" pictures=[] # 存儲(chǔ)文件夾內(nèi)所有文件的路徑(包括子目錄內(nèi)的文件) for root, dirs, files in os.walk(work_path): path = [os.path.join(root, name) for name in files] pictures.extend(path) pictures
然后將所有圖片逐個(gè)傳入表格識(shí)別接口,獲取其請(qǐng)求ID及存有識(shí)別結(jié)果的Excel文件的下載地址。在提取Excel下載路徑之前,需要先判斷識(shí)別是否完成。這是通過識(shí)別結(jié)果返回的字典中的'ret_msg'對(duì)應(yīng)的值來判斷的。只有當(dāng)它是“已完成”時(shí),才能獲得下載鏈接。此處用了while循環(huán),每隔2秒刷新一下狀態(tài),直到狀態(tài)是“已完成”時(shí),才提取鏈接。然后使用requests.get()獲取下載鏈接信息,寫入Excel文件。Excel自動(dòng)命名及下載結(jié)果如圖。
from aip import AipOcr #導(dǎo)入AipOcr模塊,用于做文字識(shí)別 import time #時(shí)間模塊 import requests #用于HTTP請(qǐng)求 APP_ID = '24800359' # 你申請(qǐng)的 API_KEY = 'PrmTtmrqkeaqhvxOPEN4eZVt'# 你申請(qǐng)的 SECRET_KEY = 'LOFpCH6wpLV7xZPG0DTcvV4x1Sqyvmk9'# 你申請(qǐng)的 client = AipOcr(APP_ID, API_KEY, SECRET_KEY) #提交識(shí)別請(qǐng)求,并儲(chǔ)存所有請(qǐng)求ID for picture in pictures: pic = open(picture,'rb') #以二進(jìn)制方式打開圖片 img = pic.read() #讀取 table = client.tableRecognitionAsync(img) #調(diào)用表格識(shí)別模塊 request_id = table['result'][0]['request_id'] #判斷識(shí)別是否完成,直到完成才根據(jù)請(qǐng)求ID獲取Excel下載路徑 result = client.getTableRecognitionResult(request_id) #通過ID獲取識(shí)別結(jié)果 while result['result']['ret_msg'] != '已完成': #如果狀態(tài)是“已完成”,才能獲取下載地址 time.sleep(2) #暫停2秒再刷新 result = client.getTableRecognitionResult(request_id) #持續(xù)刷新,直到滿足條件 download_path = result['result']['result_data'] #下載并將Excel文件名設(shè)為圖片名 excel_name = picture.split(".")[0] + ".xls" #讓excel文件的名字與圖片相同 excel = requests.get(download_path) #抓取下載鏈接 file = open(excel_name, 'wb') #新建excel文件 file.write(excel.content) #寫入excel文件并保存
總結(jié)
到此這篇關(guān)于Python調(diào)用百度AI實(shí)現(xiàn)圖片上表格識(shí)別功能的文章就介紹到這了,更多相關(guān)Python調(diào)用百度AI識(shí)別表格內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
python實(shí)現(xiàn)linux下使用xcopy的方法
這篇文章主要介紹了python實(shí)現(xiàn)linux下使用xcopy的方法,可實(shí)現(xiàn)模仿windows下的xcopy命令功能,需要的朋友可以參考下2015-06-06基于python實(shí)現(xiàn)的百度新歌榜、熱歌榜下載器(附代碼)
這篇文章主要介紹了基于python實(shí)現(xiàn)的百度新歌榜、熱歌榜下載器(附代碼),文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-08-08python3 實(shí)現(xiàn)除法結(jié)果為整數(shù)
這篇文章主要介紹了python3 實(shí)現(xiàn)除法結(jié)果為整數(shù),具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2021-03-03Python常見數(shù)據(jù)結(jié)構(gòu)之棧與隊(duì)列用法示例
這篇文章主要介紹了Python常見數(shù)據(jù)結(jié)構(gòu)之棧與隊(duì)列用法,結(jié)合實(shí)例形式簡(jiǎn)單介紹了數(shù)據(jù)結(jié)構(gòu)中棧與隊(duì)列的概念、功能及簡(jiǎn)單使用技巧,需要的朋友可以參考下2019-01-01