Python基于百度API識別并提取圖片中文字
利用百度 AI 開發(fā)平臺的 OCR 文字識別 API 識別并提取圖片中的文字。首先需注冊獲取 API 調(diào)用的 ID 和 key,步驟如下:
打開百度AI開放平臺,進(jìn)入控制臺中的文字識別應(yīng)用(需要有百度賬號)。
創(chuàng)建一個應(yīng)用,并進(jìn)入管理應(yīng)用,記下 AppID, API Key, Secrect Key,調(diào)用 API需用到。
最后安裝 python 的百度ai接口的的庫
pip install baidu-aip
以下是代碼實現(xiàn),需將所有識別的圖片放進(jìn)名為 picture 的文件夾。
#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ Created on Tue Jun 12 09:37:38 2018 利用百度api實現(xiàn)圖片文本識別 @author: XnCSD """ import glob from os import path import os from aip import AipOcr from PIL import Image def convertimg(picfile, outdir): '''調(diào)整圖片大小,對于過大的圖片進(jìn)行壓縮 picfile: 圖片路徑 outdir: 圖片輸出路徑 ''' img = Image.open(picfile) width, height = img.size while(width*height > 4000000): # 該數(shù)值壓縮后的圖片大約 兩百多k width = width // 2 height = height // 2 new_img=img.resize((width, height),Image.BILINEAR) new_img.save(path.join(outdir,os.path.basename(picfile))) def baiduOCR(picfile, outfile): """利用百度api識別文本,并保存提取的文字 picfile: 圖片文件名 outfile: 輸出文件 """ filename = path.basename(picfile) APP_ID = '******' # 剛才獲取的 ID,下同 API_KEY = '******' SECRECT_KEY = '******' client = AipOcr(APP_ID, API_KEY, SECRECT_KEY) i = open(picfile, 'rb') img = i.read() print("正在識別圖片:\t" + filename) message = client.basicGeneral(img) # 通用文字識別,每天 50 000 次免費 #message = client.basicAccurate(img) # 通用文字高精度識別,每天 800 次免費 print("識別成功!") i.close(); with open(outfile, 'a+') as fo: fo.writelines("+" * 60 + '\n') fo.writelines("識別圖片:\t" + filename + "\n" * 2) fo.writelines("文本內(nèi)容:\n") # 輸出文本內(nèi)容 for text in message.get('words_result'): fo.writelines(text.get('words') + '\n') fo.writelines('\n'*2) print("文本導(dǎo)出成功!") print() if __name__ == "__main__": outfile = 'export.txt' outdir = 'tmp' if path.exists(outfile): os.remove(outfile) if not path.exists(outdir): os.mkdir(outdir) print("壓縮過大的圖片...") // 首先對過大的圖片進(jìn)行壓縮,以提高識別速度,將壓縮的圖片保存與臨時文件夾中 for picfile in glob.glob("picture/*"): convertimg(picfile, outdir) print("圖片識別...") for picfile in glob.glob("tmp/*"): baiduOCR(picfile, outfile) os.remove(picfile) print('圖片文本提取結(jié)束!文本輸出結(jié)果位于 %s 文件中。' % outfile) os.removedirs(outdir)
到此這篇關(guān)于Python基于百度API識別并提取圖片中文字的文章就介紹到這了,更多相關(guān)Python百度API識別圖片文字內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
python 牛頓法實現(xiàn)邏輯回歸(Logistic Regression)
這篇文章主要介紹了python 牛頓法實現(xiàn)邏輯回歸(Logistic Regression),幫助大家更好的進(jìn)行機(jī)器學(xué)習(xí),感興趣的朋友可以了解下2020-10-10教你使用Python pypinyin庫實現(xiàn)漢字轉(zhuǎn)拼音
今天,發(fā)現(xiàn)了一個好玩兒的庫,叫做 “pypinyin ”,用于幫助我們實現(xiàn)漢字轉(zhuǎn)拼音,文中有非常詳細(xì)的代碼示例,對正在學(xué)習(xí)python的小伙伴們很有幫助,需要的朋友可以參考下2021-05-05Windows環(huán)境下Python3.6.8 importError: DLLload failed:找不到指定的模塊
這篇文章主要介紹了Windows環(huán)境下Python3.6.8 importError: DLLload failed:找不到指定的模塊,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-11-11Python?常用內(nèi)置模塊超詳細(xì)梳理總結(jié)
模塊是一個包含索引你定義的函數(shù)和變量的文件,其擴(kuò)展名為.py。模塊可以被其他程序引入,以使用該模塊中的函數(shù)等功能。這也是使用python標(biāo)準(zhǔn)庫的方法2022-03-03利用python3 的pygame模塊實現(xiàn)塔防游戲
這篇文章主要介紹了利用python3 的pygame模塊實現(xiàn)塔防游戲,本文給大家介紹的非常詳細(xì),具有一定的參考借鑒價值,需要的朋友可以參考下2019-12-12Python實現(xiàn)批量將符合要求的文件自動復(fù)制到新文件夾
這篇文章主要為大家詳細(xì)介紹了如何使用Python實現(xiàn)批量將文件名稱符合要求的文件自動復(fù)制到新文件夾,文中的示例代碼講解詳細(xì),有需要的小伙伴可以參考下2023-10-10