From cca3422b54afed19c461ac83a7ef5bcf04a716c7 Mon Sep 17 00:00:00 2001 From: haddyadnan Date: Thu, 19 Oct 2023 12:40:36 -0700 Subject: [PATCH] Add error msg for failed import --- deepface/detectors/FastMtcnnWrapper.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/deepface/detectors/FastMtcnnWrapper.py b/deepface/detectors/FastMtcnnWrapper.py index 2e4434019..d9b0b205a 100644 --- a/deepface/detectors/FastMtcnnWrapper.py +++ b/deepface/detectors/FastMtcnnWrapper.py @@ -6,7 +6,11 @@ def build_model(): # Optional dependency - from facenet_pytorch import MTCNN as fast_mtcnn + try: + from facenet_pytorch import MTCNN as fast_mtcnn + except ModuleNotFoundError: + print("This is an optional detector, ensure the library is installed. \ + Please install using 'pip install facenet-pytorch' ") face_detector = fast_mtcnn(image_size=160,