AWS

boto3로 Rekognition 사용하기 (python)

오늘은 집 2022. 1. 14. 17:50

0. 기본 준비

- boto3 설치

- IAM 계정

- S3버킷

- 위 S3 버킷에 올라간 이미지

 

1. 코드

# 입력해야하는 부분
photo = "사진의 이름"
bucket= "S3 버킷 이름"
S3_ACCESS_KEY = "본인 계정의 IAM 엑세스 키"
S3_SECRET_KEY = "본인 계정의 IAM 시크릿 키"

# 함수
def detect_layer():
    client=boto3.client('rekognition', 'ap-northeast-2', aws_access_key_id = S3_ACCESS_KEY, aws_secret_access_key = S3_SECRET_KEY)
    response = client.detect_labels(Image={'S3Object':{'Bucket':bucket,'Name':photo}},MaxLabels=10)
    result = []       
    for label in response['Labels']:
        label_dict = {}
        label_dict['Name'] = label['Name']
        label_dict['Confidence'] = label['Confidence']
        result.append(label_dict)
        
    return result

# 함수실행 및 결과 출력
result = detect_layer()
print(result)