guilherme.schardong/facebias
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added missing imports.

Browse Source
This commit is contained in:
Guilherme Gonçalves Schardong
2026-04-16 14:01:23 +01:00
parent 250d218f9b
commit adadf4c7fd
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/facebias/detectors/mediapipe.py
View File

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