百度图片文字识别接口封装(python)

百度SDK入口连接:https://ai.baidu.com/ai-doc/OCR/Dk3h7yf8m

"""
@File : recognition_picture_text.py
@Author : 许同学
@Time :  2020/11/8
@Describe :
    参考百度sdk:https://ai.baidu.com/ai-doc/OCR/Dk3h7yf8m
    二次封装百度图片文字识别接口
    需要安装的三方包:
        pip install baidu-aip
        pip install setuptools
@Revision history :
    None
"""

from aip import AipOcr


class RecognitionPictureText(object):
    """实现类"""
    def __init__(self):
        self.APP_ID = '***'
        self.API_KEY = '***'
        self.SECRET_KEY = '***'
        self.client = AipOcr(self.APP_ID, self.API_KEY,self. SECRET_KEY)

    def get_picture_text(self, filepath):
        with open(filepath, 'rb') as fp:
            image = fp.read()
        result = self.client.basicGeneral(image)
        text_list = []
        for i in result['words_result']:
            text_list.append(i['words'])

        return text_list


if __name__ == '__main__':
    filePath = "../data/test.png"
    r = RecognitionPictureText()
    print(r.get_picture_text(filePath))