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

move torch import from global in order to match optional dependency project philosophy

Browse Source
This commit is contained in:
aku-ato 2024-04-05 17:26:45 +02:00 committed by GitHub
parent b4c16f6dbf
commit 131a7fb576
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deepface/detectors/FastMtCnn.py
View File

@ -2,7 +2,6 @@ from typing import Any, Union, List
import cv2 import cv2
import numpy as np import numpy as np
from deepface.models.Detector import Detector, FacialAreaRegion from deepface.models.Detector import Detector, FacialAreaRegion
import torch
# Link -> https://github.com/timesler/facenet-pytorch # Link -> https://github.com/timesler/facenet-pytorch
# Examples https://www.kaggle.com/timesler/guide-to-mtcnn-in-facenet-pytorch # Examples https://www.kaggle.com/timesler/guide-to-mtcnn-in-facenet-pytorch
@ -63,6 +62,7 @@ class FastMtCnnClient(Detector):
# this is not a must dependency. do not import it in the global level. # this is not a must dependency. do not import it in the global level.
try: try:
from facenet_pytorch import MTCNN as fast_mtcnn from facenet_pytorch import MTCNN as fast_mtcnn
import torch
except ModuleNotFoundError as e: except ModuleNotFoundError as e:
raise ImportError( raise ImportError(
"FastMtcnn is an optional detector, ensure the library is installed." "FastMtcnn is an optional detector, ensure the library is installed."