github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-08 12:35:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' of https://github.com/serengil/deepface

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2021-06-26 23:13:54 +03:00
commit a35b31d242
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
deepface/detectors/RetinaFaceWrapper.py
View File

@ -1,17 +1,19 @@
from retinaface import RetinaFace #from retinaface import RetinaFace
import cv2 import cv2
def build_model(): def build_model():
from retinaface import RetinaFace
face_detector = RetinaFace.build_model() face_detector = RetinaFace.build_model()
return face_detector return face_detector
def detect_face(face_detector, img, align = True): def detect_face(face_detector, img, align = True):
from retinaface import RetinaFace
img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) #retinaface expects RGB but OpenCV read BGR
face = None face = None
img_region = [0, 0, img.shape[0], img.shape[1]] img_region = [0, 0, img.shape[0], img.shape[1]]
img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) #retinaface expects RGB but OpenCV read BGR
faces = RetinaFace.extract_faces(img_rgb, model = face_detector, align = align) faces = RetinaFace.extract_faces(img_rgb, model = face_detector, align = align)
if len(faces) > 0: if len(faces) > 0: