From e492465028c7d97acf082e3c9ed4cd96dfe9a88d Mon Sep 17 00:00:00 2001
From: Oguzhan Buyuksolak <oguzhan.buyuksolak@analog.com>
Date: Wed, 13 Dec 2023 00:46:49 +0300
Subject: [PATCH] FaceID part 2

---
 datasets/vggface2.py | 2 ++
 requirements.txt     | 1 +
 2 files changed, 3 insertions(+)

diff --git a/datasets/vggface2.py b/datasets/vggface2.py
index d431888caa..b8e8fd8f52 100644
--- 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
index ed6aa89e53..243657d545 100644
--- 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