1. 程式人生 > >python呼叫百度圖片文字識別介面

python呼叫百度圖片文字識別介面

# 登入百度api應用頁面獲取下面三相內容
APP_ID = 'xxxxx'
API_KEY = 'xxxxxxx'
SECRET_KEY = 'xxxxxxx'

class BaiduImg():
    def __init__(self, img_path):
        self.img_path = img_path # 傳入圖片地址

    """ 讀取圖片 """

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

    def img_ocr(self):
        client = AipOcr(APP_ID, API_KEY, SECRET_KEY)
        image = self.get_file_content(self.img_path)

        """ 呼叫通用文字識別, 圖片引數為本地圖片 """
        client.basicGeneral(image)
        """ 如果有可選引數 """
        options = {}
        options["language_type"] = "CHN_ENG"
        options["detect_direction"] = "true"
        options["detect_language"] = "true"
        options["probability"] = "true"
        """ 帶引數呼叫通用文字識別, 圖片引數為本地圖片 """
        bendi = client.basicGeneral(image, options)
        # print(bendi)
        return bendi  # 返回字典資料

baidu = BaiduImg('xx.png')
rst = baidu.img_ocr()
print(rst)
# {
	'log_id': 2610378325233689652,
	'direction': 0,
	'words_result_num': 4,
	'words_result': [{
		'words': '百度識圖,"鑑"你所見官網',
		'probability': {
			'variance': 0.018005,
			'average': 0.936804,
			'min': 0.486269
		}
	}, {
		'words': '百度圖片使用世界前沿的人工智慧技術為使用者頭選海量的高清美圖',
		'probability': {
			'variance': 0.016179,
			'average': 0.959505,
			'min': 0.407384
		}
	}, {
		'words': '用更流暢、更快捷、更精住的搜素體驗帶你去',
		'probability': {
			'variance': 0.012763,
			'average': 0.945982,
			'min': 0.541851
		}
	}, {
		'words': 'hitu.baidu.comr-百度快照-7519條評價',
		'probability': {
			'variance': 0.021318,
			'average': 0.892922,
			'min': 0.43175
		}
	}],
	'language': 3
}

測試圖片