함수를 사용한 이미지 분석 AWS Lambda

에 AWS Management Console 로그인하고 에서 AWS Lambda 콘솔을 엽니다 https://console.aws.amazon.com/lambda/.

함수 생성(Create function)을 선택합니다. 자세한 내용은 콘솔로 Lambda 함수 생성을 참조하세요.

다음과 같은 옵션을 선택하세요.

함수 생성을 선택하여 AWS Lambda 함수를 생성합니다.

함수 페이지에서 구성 탭을 선택합니다.

환경 변수 창에서 편집을 선택합니다.

다음 환경 변수를 추가합니다. 각 변수에 대해 환경 변수 추가를 선택한 다음 변수 키와 값을 입력합니다.

저장을 선택하여 환경 변수를 저장합니다.

권한 창의 역할 이름에서 실행 역할을 선택하여 콘솔에서 역할을 엽니다. IAM

권한 탭에서 권한 추가, 그다음 인라인 정책 생성을 선택합니다.

기존 정책을 JSON선택하고 다음 정책으로 교체합니다.

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Action": "rekognition:DetectCustomLabels",
            "Resource": "*",
            "Effect": "Allow",
            "Sid": "DetectCustomLabels"
        }
    ]
}

Next(다음)를 선택합니다.

정책 세부 정보에 정책 이름 (예: DetectCustomLabels-access) 을 입력합니다.

정책 생성을 선택합니다.

Amazon S3 버킷에 분석용 이미지를 저장하는 경우 10~14단계를 반복합니다.

1. 11단계에서는 다음 정책을 사용하십시오. Replace bucket/folder path 분석하려는 이미지의 Amazon S3 버킷 및 폴더 경로를 사용합니다.

   {
       "Version": "2012-10-17",
       "Statement": [
           {
               "Sid": "S3Access",
               "Effect": "Allow",
               "Action": "s3:GetObject",
               "Resource": "arn:aws:s3:::bucket/folder path/*"
           }
       ]
   }

2. 13단계에서는 S3Bucket-access와 같은 다른 정책 이름을 선택합니다.

명령 프롬프트를 열고 다음 명령을 입력합니다.

pip install boto3 --target python/.
zip boto3-layer.zip -r python/
                

zip 파일(boto3-layer.zip)의 이름을 기록해 두세요. 이 절차의 6단계에서 필요합니다.

에서 콘솔을 엽니다. AWS Lambda https://console.aws.amazon.com/lambda/

탐색 창에서 계층을 선택합니다.

계층 생성을 선택합니다.

Name(이름)과 Description(설명)을 입력합니다.

.zip 파일 업로드를 선택한 후 업로드를 선택합니다.

대화 상자에서 이 절차의 1단계에서 만든.zip 파일 아카이브 (boto3-layer.zip)를 선택합니다.

호환되는 런타임은 Python 3.9를 선택하세요.

생성을 선택하여 계층을 생성합니다.

탐색 창 메뉴 아이콘을 선택합니다.

탐색 창에서 함수를 선택합니다.

리소스 목록에서 1단계: AWS Lambda 함수 생성 (콘솔) 항목에서 생성한 함수를 선택합니다.

코드 탭을 선택합니다.

계층 영역에서 계층 추가를 선택합니다.

사용자 지정 계층을 선택합니다.

사용자 지정 계층에서 6단계에서 입력한 계층 이름을 선택합니다.

버전에서 계층 버전을 선택합니다. 계층 버전은 1이어야 합니다.

추가를 선택합니다.

Lambda 콘솔을 사용하지 않는 경우 다음을 수행합니다.

1. 에서 AWS Lambda 콘솔을 엽니다 https://console.aws.amazon.com/lambda/.

2. 1단계: AWS Lambda 함수 생성 (콘솔)에서 생성한 Lambda 함수를 엽니다.

코드 탭을 선택합니다.

코드 소스에서 lambda_function.py의 코드를 다음과 같이 바꾸세요.

# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0

"""
Purpose
An AWS lambda function that analyzes images with an the Amazon Rekognition
Custom Labels model.
"""
import json
import base64
from os import environ
import logging
import boto3

from botocore.exceptions import ClientError

# Set up logging.
logger = logging.getLogger(__name__)

# Get the model ARN and confidence.
model_arn = environ['MODEL_ARN']
min_confidence = int(environ.get('CONFIDENCE', 50))

# Get the boto3 client.
rek_client = boto3.client('rekognition')


def lambda_handler(event, context):
    """
    Lambda handler function
    param: event: The event object for the Lambda function.
    param: context: The context object for the lambda function.
    return: The labels found in the image passed in the event
    object.
    """

    try:

        # Determine image source.
        if 'image' in event:
            # Decode the image
            image_bytes = event['image'].encode('utf-8')
            img_b64decoded = base64.b64decode(image_bytes)
            image = {'Bytes': img_b64decoded}


        elif 'S3Object' in event:
            image = {'S3Object':
                     {'Bucket':  event['S3Object']['Bucket'],
                      'Name': event['S3Object']['Name']}
                     }

        else:
            raise ValueError(
                'Invalid source. Only image base 64 encoded image bytes or S3Object are supported.')


        # Analyze the image.
        response = rek_client.detect_custom_labels(Image=image,
            MinConfidence=min_confidence,
            ProjectVersionArn=model_arn)

        # Get the custom labels
        labels = response['CustomLabels']

        lambda_response = {
            "statusCode": 200,
            "body": json.dumps(labels)
        }

    except ClientError as err:
        error_message = f"Couldn't analyze image. " + \
            err.response['Error']['Message']

        lambda_response = {
            'statusCode': 400,
            'body': {
                "Error": err.response['Error']['Code'],
                "ErrorMessage": error_message
            }
        }
        logger.error("Error function %s: %s",
            context.invoked_function_arn, error_message)

    except ValueError as val_error:
        lambda_response = {
            'statusCode': 400,
            'body': {
                "Error": "ValueError",
                "ErrorMessage": format(val_error)
            }
        }
        logger.error("Error function %s: %s",
            context.invoked_function_arn, format(val_error))

    return lambda_response

Lambda 함수를 배포하려면 배포를 선택합니다.

lambda:InvokeFunction 권한이 있는지 확인합니다. 다음 정책을 사용할 수 있습니다.

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Sid": "InvokeLambda",
            "Effect": "Allow",
            "Action": "lambda:InvokeFunction",
            "Resource": "ARN for lambda function"
        }
    ]
}

Lambda 콘솔의 함수 개요에서 Lambda 함수 함수에 ARN 대한 정보를 얻을 수 있습니다.

액세스 권한을 제공하려면 사용자, 그룹 또는 역할에 권한을 추가하세요:

AWS SDKPython을 설치하고 구성합니다. 자세한 내용은 4단계: 설정 AWS CLI 그리고 AWS SDKs 단원을 참조하십시오.

1단계: AWS Lambda 함수 생성 (콘솔)의 7단계에서 지정한 모델을 시작합니다.

다음 코드를 client.py 이름의 파일에 저장합니다.

# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0

"""
Purpose
Test code for running the Amazon Rekognition Custom Labels Lambda
function example code.
"""

import argparse
import logging
import base64
import json
import boto3

from botocore.exceptions import ClientError


logger = logging.getLogger(__name__)


def analyze_image(function_name, image):
    """Analyzes an image with an AWS Lambda function.
    :param image: The image that you want to analyze.
    :return The status and classification result for
    the image analysis.
    """

    lambda_client = boto3.client('lambda')

    lambda_payload = {}

    if image.startswith('s3://'):
        logger.info("Analyzing image from S3 bucket: %s", image)
        bucket, key = image.replace("s3://", "").split("/", 1)
        s3_object = {
            'Bucket': bucket,
            'Name': key
        }
        lambda_payload = {"S3Object": s3_object}

    # Call the lambda function with the image.
    else:
        with open(image, 'rb') as image_file:
            logger.info("Analyzing local image image: %s ", image)
            image_bytes = image_file.read()
            data = base64.b64encode(image_bytes).decode("utf8")

            lambda_payload = {"image": data}

    response = lambda_client.invoke(FunctionName=function_name,
                                    Payload=json.dumps(lambda_payload))

    return json.loads(response['Payload'].read().decode())


def add_arguments(parser):
    """
    Adds command line arguments to the parser.
    :param parser: The command line parser.
    """

    parser.add_argument(
        "function", help="The name of the AWS Lambda function that you want " \
        "to use to analyze the image.")
    parser.add_argument(
        "image", help="The local image that you want to analyze.")


def main():
    """
    Entrypoint for script.
    """
    try:
        logging.basicConfig(level=logging.INFO,
                            format="%(levelname)s: %(message)s")

        # Get command line arguments.
        parser = argparse.ArgumentParser(usage=argparse.SUPPRESS)
        add_arguments(parser)
        args = parser.parse_args()

        # Get analysis results.
        result = analyze_image(args.function, args.image)
        status = result['statusCode']

        if status == 200:
            labels = result['body']
            labels = json.loads(labels)
            print(f"There are {len(labels)} labels in the image.")
            for custom_label in labels:
                confidence = int(round(custom_label['Confidence'], 0))
                print(
                    f"Label: {custom_label['Name']}: Confidence: {confidence}%")
        else:
            print(f"Error: {result['statusCode']}")
            print(f"Message: {result['body']}")

    except ClientError as error:
        logging.error(error)
        print(error)


if __name__ == "__main__":
    main()

코드를 실행합니다. 명령줄 인수에 Lambda 함수 이름과 분석하려는 이미지를 제공하세요. 로컬 이미지에 대한 경로 또는 Amazon S3 버킷에 저장된 이미지에 대한 S3 경로를 제공할 수 있습니다. 예제:

python client.py function_name s3://bucket/path/image.jpg

이미지가 Amazon S3 버킷에 있는 경우, 1단계: AWS Lambda 함수 생성 (콘솔)의 15단계에서 지정한 것과 동일한 버킷인지 확인하세요.

성공하면 이미지에서 찾은 레이블 목록이 출력됩니다. 레이블이 반환되지 않는 경우 1단계: AWS Lambda 함수 생성 (콘솔)의 7단계에서 설정한 신뢰도 값을 낮추는 것이 좋습니다.

Lambda 함수 사용을 마쳤지만 다른 애플리케이션에서 해당 모델을 사용하지 않는 경우 모델을 중지하세요. 다음에 Lambda 함수를 사용하려는 경우 모델을 시작해야 한다는 점을 기억하십시오.