debug

src/utils/crop_face_single.py

@@ -3,7 +3,10 @@ import cv2
 
 
 def crop_face(img, lmk_extractor, expand=1.5):
+    print('****=====1======')
     result = lmk_extractor(img)  # cv2 BGR
+    
+    print('****=====2======')
 
     if result is None:
         return None

src/utils/mp_utils.py

@@ -1,4 +1,5 @@
 import os
+import sys
 import numpy as np
 import cv2
 import time
@@ -11,7 +12,7 @@ from mediapipe import solutions
 from mediapipe.framework.formats import landmark_pb2
 from mediapipe.tasks import python
 from mediapipe.tasks.python import vision
-from . import face_landmark
+from src.utils import face_landmark
 
 CUR_DIR = os.path.dirname(__file__)
 
@@ -31,35 +32,36 @@ class LMKExtractor():
         self.last_ts = 0
         self.frame_ms = int(1000 / FPS)
 
-        det_base_options = python.BaseOptions(model_asset_path=os.path.join(CUR_DIR, 'mp_models/blaze_face_short_range.tflite'))
-        det_options = vision.FaceDetectorOptions(base_options=det_base_options)
-        self.det_detector = vision.FaceDetector.create_from_options(det_options)
+        # det_base_options = python.BaseOptions(model_asset_path=os.path.join(CUR_DIR, 'mp_models/blaze_face_short_range.tflite'))
+        # det_options = vision.FaceDetectorOptions(base_options=det_base_options)
+        # self.det_detector = vision.FaceDetector.create_from_options(det_options)
                 
 
     def __call__(self, img):
+        print('///=====1======')
         frame = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
         image = mp.Image(image_format=mp.ImageFormat.SRGB, data=frame)
-        t0 = time.time()
-        if self.mode == mp.tasks.vision.FaceDetectorOptions.running_mode.VIDEO:
-            det_result = self.det_detector.detect(image)
-            if len(det_result.detections) != 1:
-                return None
-            self.last_ts += self.frame_ms
-            try:
-                detection_result, mesh3d = self.detector.detect_for_video(image, timestamp_ms=self.last_ts)
-            except:
-                return None
-        elif self.mode == mp.tasks.vision.FaceDetectorOptions.running_mode.IMAGE:
+        # t0 = time.time()
+        # if self.mode == mp.tasks.vision.FaceDetectorOptions.running_mode.VIDEO:
+        #     det_result = self.det_detector.detect(image)
+        #     if len(det_result.detections) != 1:
+        #         return None
+        #     self.last_ts += self.frame_ms
+        #     try:
+        #         detection_result, mesh3d = self.detector.detect_for_video(image, timestamp_ms=self.last_ts)
+        #     except:
+        #         return None
+        # elif self.mode == mp.tasks.vision.FaceDetectorOptions.running_mode.IMAGE:
             # det_result = self.det_detector.detect(image)
 
             # if len(det_result.detections) != 1:
             #     return None
-            try:
-                detection_result, mesh3d = self.detector.detect(image)
-            except:
-                return None
+        try:
+            detection_result, mesh3d = self.detector.detect(image)
+        except:
+            return None
             
-        
+        print('///=====2======')
         bs_list = detection_result.face_blendshapes
         if len(bs_list) == 1:
             bs = bs_list[0]