[PYTHON] Anime face detection with OpenCV

import os
import cv2

#Create a classifier based on the feature file
classifier = cv2.CascadeClassifier('lbpcascade_animeface.xml')

#Face detection
image = cv2.imread('newGame.jpg')
#Speed up processing with grayscale
gray_image = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)
faces = classifier.detectMultiScale(gray_image)

[[485 148 134 134]
 [456 313 193 193]
 [380  58  98  98]
 [649 227 127 127]
 [373 245 108 108]
 [637  54 104 104]]

#Create directory
output_dir = 'faces'
if not os.path.exists(output_dir):
    os.makedirs(output_dir)
    
for i, (x,y,w,h) in enumerate(faces):
    #Cut out the face one by one
    face_image = image[y:y+h, x:x+w]
    output_path = os.path.join(output_dir,'{0}.jpg'.format(i))
    cv2.imwrite(output_path,face_image)
    
cv2.imwrite('face.jpg',image)

for x,y,w,h in faces:
    #Draw a square
    cv2.rectangle(image, (x,y), (x+w,y+h), color=(0,0,255), thickness=3)
    
cv2.imwrite('faces.jpg',image)

Recommended Posts