celebX / face_recognition.py
Salahidine Lemaachi
update
08b7b53
import cv2
import numpy as np
import os
dirname = os.path.dirname(__file__)
#filename = os.path.join(dirname, 'haarcascade_frontalface_default.xml')
filename = os.path.join(dirname, 'haarcascade_frontalface_default.xml')
def conv_gris(img):
"""conversion image grise"""
img_gris=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
return img_gris
def hog(img):
img_gris = conv_gris(img)
face_cascade = cv2.CascadeClassifier(filename)
faces = face_cascade.detectMultiScale(img_gris, 1.2, 1)
return faces
def haar_cascade(img):
faces = hog(img)
for (x, y, w, h) in faces:
cv2.rectangle(img, (x, y), (x+w, y+h), (255, 0, 0), 2)
return img
if __name__ == "__main__" :
url = "./Data/Abbi.jpg"
Img = cv2.imread(url, 1)
img = haar_cascade(Img)
cv2.imshow('image_test', img)
cv2.waitKey()
cv2.destroyWindow()