pythonはBaiduのAPIを利用して画像文字(マルチスレッド版)を識別する。
#!/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
from queue import Queue
import threading
import datetime
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(ts_queue):
""" api ,
picfile:
outfile:
"""
while not ts_queue.empty():
picfile = ts_queue.get()
filename = path.basename(picfile)
outfile = 'D:\Study\pythonProject\scrapy\IpProxy\port_zidian.txt'
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('
')
# fo.writelines("+" * 60 + '
')
# fo.writelines(" :\t" + filename + "
" * 2)
# fo.writelines(" :
")
# #
# for text in message.get('words_result'):
# fo.writelines(text.get('words') + '
')
# fo.writelines('
' * 2)
os.remove(filename)
print(" !")
except:
print(' ')
print(" !")
print()
def duqu_tupian(dir):
ts_queue = Queue(10000)
outdir = dir
# if path.exists(outfile):
# os.remove(outfile)
if not path.exists(outdir):
os.mkdir(outdir)
print(" ...")
# , ,
try:
for picfile in glob.glob(r"D:\Study\pythonProject\scrapy\IpProxy\ \*"):
convertimg(picfile, outdir)
print(" ...")
for picfile in glob.glob("tmp/*"):
ts_queue.put(picfile)
#baiduOCR(picfile, outfile)
#os.remove(picfile)
print(' ! 。' )
#os.removedirs(outdir)
return ts_queue
except:
print(' ')
if __name__ == "__main__":
start = datetime.datetime.now().replace(microsecond=0)
t = 'tmp'
s = duqu_tupian(t)
threads = []
for i in range(100):
t = threading.Thread(target=baiduOCR, name='th-' + str(i), kwargs={'ts_queue': s})
threads.append(t)
for t in threads:
t.start()
for t in threads:
t.join()
end = datetime.datetime.now().replace(microsecond=0)
print(' :' + str(end - start))
スピードが速くて、正確率は99%です。100里で間違いがあります。実測、1500枚の画像を識別しますか?それとも小さいピクチャの検証コードの大きさですか?高清、使う時30秒、150枚を識別することができません。14枚ぐらい間違えました。 全体的に速くて、文字化けなどはありません。
以上はpythonを利用して、BaiduのAPIで画像文字(マルチスレッド版)の詳細を識別します。pythonに関する画像文字の識別に関する資料は他の関連記事に注目してください。