From 16f1ac0d48d652684f79e943431fa18160f1a4b1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=9Eefik=20Serangil?= <sefik.serengil@softtech.com.tr>
Date: Sat, 5 Sep 2020 22:53:47 +0300
Subject: [PATCH] unit tests for detector backends

---
 tests/unit_tests.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/tests/unit_tests.py b/tests/unit_tests.py
index 077f2b2..da647e3 100644
--- a/tests/unit_tests.py
+++ b/tests/unit_tests.py
@@ -1,5 +1,7 @@
 from deepface import DeepFace
+from deepface.commons import functions
 import json
+import time
 
 import os
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
@@ -211,5 +213,22 @@ df = DeepFace.find("dataset/img1.jpg", db_path = "dataset", model_name = 'Ensemb
 
 print(df)
 
+#-----------------------------------
+print("--------------------------")
+
+print("Different face detector backends")
+
+backends = ['opencv', 'ssd', 'dlib', 'mtcnn']
+
+for backend in backends:
+	
+	tic = time.time()
+	
+	processed_img = functions.preprocess_face(img = "dataset/img11.jpg", detector_backend = backend)
+	
+	toc = time.time()
+	
+	print("Backend ", backend, " is done in ", toc-tic," seconds")
+
 #-----------------------------------
 print("--------------------------")
\ No newline at end of file