github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 20:15:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

retinaface is not a must dependency

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2021-06-25 16:01:47 +03:00 committed by GitHub
parent dcd883f223
commit 2c2aaa0ede
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
deepface/detectors/RetinaFaceWrapper.py
View File

@ -1,17 +1,19 @@
from retinaface import RetinaFace
#from retinaface import RetinaFace
import cv2
def build_model():
from retinaface import RetinaFace
face_detector = RetinaFace.build_model()
return face_detector
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
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)
if len(faces) > 0: