基于PYQT5及百度aip的文字识别程序。

基于PYQT5及百度aip的文字识别程序。

#coding:utf-8
from PyQt5.Qt import *
import sys
from aip import AipOcr
class App(QWidget):
    def __init__(self):
        super().__init__()
        self.author='薛辉辉'
        self.setWindowTitle('文字识别程序 1.0   by:xuehuihui')
        # self.resize(500,500)
        self.setFixedSize(500,500)
        self.btn=QPushButton('打开',self)
        self.btn.move(20,20)
        self.btn.resize(80,30)
        self.btn.clicked.connect(self.openfiles)
        self.text1=QTextEdit(self)
        self.text1.move(20,50)
        self.text1.resize(460,430)
    def openfiles(self):
        self.text1.clear()
        self.a,b=QFileDialog.getOpenFileNames(self, "多文件选择","./","All Files (*.jpg *.gif *.png *.jpeg);;JPG Files (*.jpg);;PNG Files (*.png)")
        print(self.a,b)
        for i in self.a:
            self.baidu(i)
    def baidu(self,path):
        APP_ID = '自行填写'
        API_KEY = '自行填写'
        SECRET_KEY = '自行填写'
        magnification = 0.6
        client = AipOcr(APP_ID, API_KEY, SECRET_KEY)

        def get_file_content(filePath):
            with open(filePath, 'rb') as fp:
                return fp.read()

        image = get_file_content(path)
        options = {}
        options["recognize_granularity"] = "big"
        options["language_type"] = "CHN_ENG"
        options["detect_direction"] = "true"
        options["detect_language"] = "true"
        options["vertexes_location"] = "true"
        options["probability"] = "false"
        a = client.general(image, options)
        word = '\n'
        loc = []
        word_len = a['words_result'][0]['finegrained_vertexes_location'][1]['x'] - \
                   a['words_result'][0]['finegrained_vertexes_location'][0]['x']
        for i in range(0, a['words_result_num']):
            if i == a['words_result_num'] - 1:
                word = word + a['words_result'][i]['words']
                break
            else:
                if a['words_result'][i]['location']['left'] > word_len * magnification + \
                        a['words_result'][i + 1]['location']['left'] and a['words_result'][i]['location']['width'] < \
                        a['words_result'][i + 1]['location']['width']:
                    word = word + '\n    ' + a['words_result'][i]['words']
                else:
                    word = word + a['words_result'][i]['words']
        print(word)
        self.text1.append(word)
if __name__ == '__main__':
    app=QApplication(sys.argv)
    window=App()
    window.show()
    sys.exit(app.exec_())