github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 12:05:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add error msg for failed import

Browse Source
This commit is contained in:
haddyadnan 2023-10-19 12:40:36 -07:00
parent c53366b75a
commit cca3422b54
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deepface/detectors/FastMtcnnWrapper.py
View File

@ -6,7 +6,11 @@ from deepface.detectors import FaceDetector
def build_model(): def build_model():
# Optional dependency # 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, face_detector = fast_mtcnn(image_size=160,