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

muthuvel-murugan · Jun 26, 2021 · a35b31d · a35b31d
2 parents 0fc1672 + 2c2aaa0
commit a35b31d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/deepface/detectors/RetinaFaceWrapper.py b/deepface/detectors/RetinaFaceWrapper.py
@@ -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: