diff --git a/deepface/basemodels/ArcFace.py b/deepface/basemodels/ArcFace.py index 0ec0834..6015b4b 100644 --- a/deepface/basemodels/ArcFace.py +++ b/deepface/basemodels/ArcFace.py @@ -1,7 +1,5 @@ -from typing import List import os import gdown -import numpy as np from deepface.commons import package_utils, folder_utils from deepface.commons.logger import Logger from deepface.models.FacialRecognition import FacialRecognition diff --git a/deepface/basemodels/DeepID.py b/deepface/basemodels/DeepID.py index 65a797f..f2168ff 100644 --- a/deepface/basemodels/DeepID.py +++ b/deepface/basemodels/DeepID.py @@ -1,7 +1,5 @@ -from typing import List import os import gdown -import numpy as np from deepface.commons import package_utils, folder_utils from deepface.commons.logger import Logger from deepface.models.FacialRecognition import FacialRecognition diff --git a/deepface/basemodels/Dlib.py b/deepface/basemodels/Dlib.py index 225aaf8..778b2e1 100644 --- a/deepface/basemodels/Dlib.py +++ b/deepface/basemodels/Dlib.py @@ -25,8 +25,9 @@ class DlibClient(FacialRecognition): def forward(self, img: np.ndarray) -> List[float]: """ - Find embeddings with Dlib model - Overwritten because it is different than regular models + Find embeddings with Dlib model. + This model necessitates the override of the forward method + because it is not a keras model. Args: img (np.ndarray): pre-loaded image in BGR Returns diff --git a/deepface/basemodels/Facenet.py b/deepface/basemodels/Facenet.py index e3183d4..63852f5 100644 --- a/deepface/basemodels/Facenet.py +++ b/deepface/basemodels/Facenet.py @@ -1,7 +1,5 @@ -from typing import List import os import gdown -import numpy as np from deepface.commons import package_utils, folder_utils from deepface.commons.logger import Logger from deepface.models.FacialRecognition import FacialRecognition diff --git a/deepface/basemodels/FbDeepFace.py b/deepface/basemodels/FbDeepFace.py index dc3faba..630ad6f 100644 --- a/deepface/basemodels/FbDeepFace.py +++ b/deepface/basemodels/FbDeepFace.py @@ -1,8 +1,6 @@ -from typing import List import os import zipfile import gdown -import numpy as np from deepface.commons import package_utils, folder_utils from deepface.commons.logger import Logger from deepface.models.FacialRecognition import FacialRecognition diff --git a/deepface/basemodels/GhostFaceNet.py b/deepface/basemodels/GhostFaceNet.py index 5a65102..d6347c5 100644 --- a/deepface/basemodels/GhostFaceNet.py +++ b/deepface/basemodels/GhostFaceNet.py @@ -1,10 +1,8 @@ # built-in dependencies import os -from typing import List # 3rd party dependencies import gdown -import numpy as np import tensorflow as tf # project dependencies diff --git a/deepface/basemodels/OpenFace.py b/deepface/basemodels/OpenFace.py index 4d410ab..a21dd08 100644 --- a/deepface/basemodels/OpenFace.py +++ b/deepface/basemodels/OpenFace.py @@ -1,8 +1,6 @@ -from typing import List import os import gdown import tensorflow as tf -import numpy as np from deepface.commons import package_utils, folder_utils from deepface.commons.logger import Logger from deepface.models.FacialRecognition import FacialRecognition diff --git a/deepface/basemodels/SFace.py b/deepface/basemodels/SFace.py index 6fd7649..b5cf649 100644 --- a/deepface/basemodels/SFace.py +++ b/deepface/basemodels/SFace.py @@ -28,7 +28,8 @@ class SFaceClient(FacialRecognition): def forward(self, img: np.ndarray) -> List[float]: """ Find embeddings with SFace model - Overwritten because it is different than regular models + This model necessitates the override of the forward method + because it is not a keras model. Args: img (np.ndarray): pre-loaded image in BGR Returns