diff --git a/src/facebias/detectors/mediapipe.py b/src/facebias/detectors/mediapipe.py index 2d52ac1..9aeb43e 100755 --- a/src/facebias/detectors/mediapipe.py +++ b/src/facebias/detectors/mediapipe.py @@ -1,9 +1,11 @@ #!/usr/bin/env python # -*- coding: utf-8 -*- +import numpy as np from facebias import FaceBox from facebias.detectors import BaseFaceDetector, NoFaceDetectedError +import mediapipe as mp from mediapipe.tasks.python import BaseOptions, vision @@ -55,9 +57,10 @@ class MediapipeDetector(BaseFaceDetector): if __name__ == '__main__': - import cv2 from pathlib import Path + import cv2 + TEST_IM_PATH = Path("data/frll_neutral_front/001_03.jpg") DETECTOR_PATH = Path("models/blaze_face_short_range.tflite")