dghs_imgutils_anime_face_detection_and_cropping.py

1from imgutils.detect import detect_faces
2from imgutils.data import load_image
3
4# Load an image
5image_path = 'input_image.jpg'
6image = load_image(image_path)
7
8# Detect anime faces in the image
9# This returns a list of tuples: (bbox, type, score)
10# bbox is [x0, y0, x1, y1]
11results = detect_faces(image)
12
13for i, (bbox, type_, score) in enumerate(results):
14    print(f"Face {i+1}: box={bbox}, type={type_}, score={score:.2f}")
15    
16    # Crop the face from the original image using the bounding box
17    face_img = image.crop(bbox)
18    face_img.save(f'face_{i+1}.png')