ASJPush/Object/utils/AmazonRekognitionUtil.py

# -*- 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