unit tests updated

- FaceNet512 added into unit test cases - pylint disable added for borad except
zt449569708 · Dec 8, 2023 · 39e29c0 · 39e29c0
1 parent c69f9fc
commit 39e29c0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit_tests.py b/tests/unit_tests.py
@@ -9,7 +9,7 @@
 
 logger = Logger()
 
-# pylint: disable=consider-iterating-dictionary
+# pylint: disable=consider-iterating-dictionary,broad-except
 
 logger.info("-----------------------------------------")
 
@@ -45,7 +45,7 @@ def evaluate(condition):
 # ------------------------------------------------
 
 detectors = ["opencv", "mtcnn"]
-models = ["VGG-Face", "Facenet", "ArcFace"]
+models = ["VGG-Face", "Facenet", "Facenet512", "ArcFace"]
 metrics = ["cosine", "euclidean", "euclidean_l2"]
 
 dataset = [