Cloud Function code for Video Intelligence API pose detection on a Storage Bucket Trigger

main.py

from google.cloud import videointelligence_v1p3beta1 as videointelligence

client = videointelligence.VideoIntelligenceServiceClient()

OUTPUT_BUCKET = 'gs://YOUR OUTPUT BUCKET NAME/'

# configure person detection
config = videointelligence.types.PersonDetectionConfig(
    include_bounding_boxes=True, # include bounding box round whole body
    include_attributes=False, # include cloths info
    include_pose_landmarks=True, # include body joints info
)

video_context = videointelligence.types.VideoContext(person_detection_config=config)

def hello_gcs(event, context):
    print(event)

    gcs_uri = 'gs://' + event['bucket'] + '/' + event['name']

    just_file_name = event['name'].split('.')[0]

    operation = client.annotate_video(
        input_uri=gcs_uri,
        features=[videointelligence.enums.Feature.PERSON_DETECTION],
        video_context=video_context,
        output_uri = OUTPUT_BUCKET + just_file_name + '.json'
    )
    print("\nProcessing video for person detection annotations.")
    

requirnments.txt

google-cloud-videointelligence