wko_anime-face-similarity/test.ipynb

# import required libraries
import cv2

# read the input image
img = cv2.imread('data/UAM-Andrzej-Wójtowicz.jpg')

# convert to grayscale of each frames
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# read the haarcascade to detect the faces in an image
face_cascade = cv2.CascadeClassifier('haarcascades/haarcascade_frontalface_default.xml')

# detects faces in the input image
faces = face_cascade.detectMultiScale(gray, 1.3, 4)
print('Number of detected faces:', len(faces))

# loop over all detected faces
if len(faces) > 0:
   for i, (x, y, w, h) in enumerate(faces):
      # To draw a rectangle in a face
      cv2.rectangle(img, (x, y), (x + w, y + h), (255, 255, 255), 2)
      face = img[y:y + h, x:x + w]
      cv2.imshow("Cropped Face", face)
      cv2.imwrite(f'face{i}.jpg', face)
      print(f"face{i}.jpg is saved")
 
# display the image with detected faces
cv2.imshow("image", img)

Number of detected faces: 1
face0.jpg is saved