dghs_imgutils_detect_person_and_crop_bounding_boxes.py

1from imgutils.detect import detect_person
2from imgutils.data import load_image
3
4# Load an image
5image_file = 'input_image.jpg'
6image = load_image(image_file)
7
8# Detect persons/characters in the image
9# This returns a list of tuples: (bbox, label, score)
10results = detect_person(image)
11
12# Print results and crop the first detected person
13for i, (bbox, label, score) in enumerate(results):
14    print(f"Detected {label} with confidence {score:.2f} at {bbox}")
15    
16    # Crop the detected area
17    # bbox format is (x1, y1, x2, y2)
18    cropped_image = image.crop(bbox)
19    cropped_image.save(f'person_{i}.png')