FaceID part 2

oguzhanbsolak · Dec 12, 2023 · e492465 · e492465
1 parent 03c1e99
commit e492465
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/datasets/vggface2.py b/datasets/vggface2.py
@@ -24,6 +24,7 @@
 
 import cv2
 import kornia.geometry.transform as GT
+import onnxruntime
 from hawk_eyes.face import RetinaFace
 from PIL import Image
 from skimage import transform as trans
@@ -99,6 +100,7 @@ def __extract_gt(self):
         """
         Extracts the ground truth from the dataset
         """
+        onnxruntime.set_default_logger_severity(3) # suppress onnxruntime warnings
         retina = RetinaFace(model_name='retina_l', conf=0.5)
         img_paths = list(glob.glob(os.path.join(self.d_path + '/**/', '*.jpg'), recursive=True))
         nf_number = 0

diff --git a/requirements.txt b/requirements.txt
@@ -25,4 +25,5 @@ hawk-eyes==2.1.2
 onnx==1.15.0
 onnxruntime-gpu==1.10.0
 imutils==0.5.4
+kornia==0.6.8
 -e distiller