# -*- coding: utf-8 -*- """ @Author : Rocky @Time : 2022/12/12 14:28 @File :AmazonRekognitionUtil.py """ import boto3 class AmazonRekognitionUtil: # doc: https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/rekognition.html def __init__(self): # 全部使用美东服务 self.region_name = 'us-east-1' self.aws_access_key_id = 'AKIA2E67UIMD6JD6TN3J' self.aws_secret_access_key = '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' self.client = boto3.client( 'rekognition', region_name=self.region_name, aws_access_key_id=self.aws_access_key_id, aws_secret_access_key=self.aws_secret_access_key, ) def detect_labels(self, image): """ 识别图片标签 @param image: 图片二进制文件 @return: rekognition_res """ rekognition_res = self.client.detect_labels( Image={'Bytes': image}, MaxLabels=50, MinConfidence=80 ) try: assert rekognition_res['ResponseMetadata']['HTTPStatusCode'] == 200 except AssertionError: return {} return rekognition_res