python實現百度OCR圖片識別

一、直接上代碼

import base64
import requests

class CodeDemo:
    def __init__(self,AK,SK,code_url,img_path):
        self.AK=AK
        self.SK=SK
        self.code_url=code_url
        self.img_path=img_path
        self.access_token=self.get_access_token()

    def get_access_token(self):
        token_host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={ak}&client_secret={sk}'.format(ak=self.AK,sk=self.SK)
        header={'Content-Type': 'application/json; charset=UTF-8'}
        response=requests.post(url=token_host,headers=header)
        content = response.json()
        access_token=content.get("access_token")
        return access_token

    def getCode(self):
        header = {
            "Content-Type": "application/x-www-form-urlencoded"
        }
        def read_img():
            with open(self.img_path, "rb")as f:
                return base64.b64encode(f.read()).decode()

        image = read_img()
        response=requests.post(url=self.code_url,data={"image":image,"access_token":self.access_token},headers=header)
        return response.json()

if __name__ == '__main__':
    AK = "" # 官網獲取的AK
    SK = "" # 官網獲取的SK
    code_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/accurate" # 百度圖片識別接口地址
    img_path=r"" # 識別圖片的地址

    code_obj=CodeDemo(AK=AK,SK=SK,code_url=code_url,img_path=img_path)
    res=code_obj.getCode()
    code=res.get("words_result")[0].get("words")
    print(res)
    print(code)