1. 程式人生 > 實用技巧 >python 利用百度API批量識別圖片文字【1】

python 利用百度API批量識別圖片文字【1】

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Jun 12 09:37:38 2018
利用百度api實現圖片文字識別
@author: XnCSD
"""

import glob
from os import path
import os
from aip import AipOcr
from PIL import Image


def convertimg(picfile, outdir):
    '''調整圖片大小,對於過大的圖片進行壓縮
    picfile:    圖片路徑
    outdir:    圖片輸出路徑
    
''' img = Image.open(picfile) width, height = img.size while (width * height > 4000000): # 該數值壓縮後的圖片大約 兩百多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() try: filename1 = filename.split('.')[0] filename1 = ''.join(filename1) with open(outfile, 'a+') as fo: for text in message.get('words_result'): fo.writelines('\'' + filename1 + '\'' + ':' + text.get('words') + ',') fo.writelines('\n') # fo.writelines("+" * 60 + '\n') # fo.writelines("識別圖片:\t" + filename + "\n" * 2) # fo.writelines("文字內容:\n") # # 輸出文字內容 # for text in message.get('words_result'): # fo.writelines(text.get('words') + '\n') # fo.writelines('\n' * 2) except: print('識別失敗') print("文字匯出成功!") print() if __name__ == "__main__": outfile = 'port_zidian.txt' outdir = 'tmp' if path.exists(outfile): os.remove(outfile) if not path.exists(outdir): os.mkdir(outdir) print("壓縮過大的圖片...") #首先對過大的圖片進行壓縮,以提高識別速度,將壓縮的圖片儲存與臨時資料夾中 try: for picfile in glob.glob("埠/*"): convertimg(picfile, outdir) print("圖片識別...") for picfile in glob.glob("tmp/*"): baiduOCR(picfile, outfile) os.remove(picfile) print('圖片文字提取結束!文字輸出結果位於 %s 檔案中。' % outfile) os.removedirs(outdir) except: print('失敗')

速度挺快的,準確率99%,確切的說 100張高清的圖片給識別 必會亂識別一張,還是最簡單的數字。

但碰到多圖片慢了。