insightface_face_detection_with_landmarks_and_bounding_boxes.py

1import cv2
2import numpy as np
3import insightface
4from insightface.app import FaceAnalysis
5from insightface.data import get_image as ins_get_image
6
7# Initialize the FaceAnalysis app
8# 'providers' can be ['CUDAExecutionProvider', 'CPUExecutionProvider']
9app = FaceAnalysis(providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
10app.prepare(ctx_id=0, det_size=(640, 640))
11
12# Load a sample image (using insightface's built-in helper or cv2.imread)
13img = ins_get_image('t1')
14
15# Perform detection and recognition
16faces = app.get(img)
17
18# Draw results on the image
19rimg = app.draw_on(img, faces)
20
21# Save or display the result
22cv2.imwrite("./output.jpg", rimg)
23print(f"Found {len(faces)} faces. Result saved to output.jpg")