detect_labels.py

import boto3


# From Alex Casalboni: https://gist.github.com/alexcasalboni/0f21a1889f09760f8981b643326730ff
def detect_labels(bucket, key, max_labels=10, min_confidence=95, region=REGION):
    
    # I have separate profiles in aws so I have to define one here
    # in order to authenticate myself to my account
    # If you do not have multiple profiles, just remove the argument
    session = boto3.Session(profile_name='<profile bname>')
    
    rekognition = session.client("rekognition", region)
    response = rekognition.detect_labels(
        Image={
            "S3Object": {
                "Bucket": bucket,
                "Name": key,
            }
        },
        MaxLabels=max_labels,
        MinConfidence=min_confidence,
    )
    return response.get('Labels', None)