Switched to mediapipe

align.py

@@ -1,5 +1,8 @@
 import numpy as np
-import face_recognition
+import mediapipe as mp
+
+mp_face_detection = mp.solutions.face_detection
+face_detection = mp_face_detection.FaceDetection(min_detection_confidence=0.5, model_selection=0)
 
 def get_locations(numpy_array,model_type="dlib"):
     '''

requirements.txt

@@ -3,4 +3,5 @@ onnxruntime-gpu
 opencv-python
 numpy
 #face_recognition
-dlib
+#dlib
+mediapipe

zombie.py

@@ -67,7 +67,7 @@ def transition(path,model):
     return get_concat_h(square,zombie)
 
 def transition_onnx(rgb_image,model):
-    square,_ = align.align(rgb_image,enable_padding=False)
+    square,_ = align.align(rgb_image,enable_padding=False,model_type="mediapipe")
     # if square is None:
     #     return None