unused imports removed and docstring detailed for sface and dlib

This commit is contained in:
Sefik Ilkin Serengil 2024-03-31 09:25:13 +01:00
parent 816f945136
commit 504dc3c66d
8 changed files with 5 additions and 15 deletions

View File

@ -1,7 +1,5 @@
from typing import List
import os import os
import gdown import gdown
import numpy as np
from deepface.commons import package_utils, folder_utils from deepface.commons import package_utils, folder_utils
from deepface.commons.logger import Logger from deepface.commons.logger import Logger
from deepface.models.FacialRecognition import FacialRecognition from deepface.models.FacialRecognition import FacialRecognition

View File

@ -1,7 +1,5 @@
from typing import List
import os import os
import gdown import gdown
import numpy as np
from deepface.commons import package_utils, folder_utils from deepface.commons import package_utils, folder_utils
from deepface.commons.logger import Logger from deepface.commons.logger import Logger
from deepface.models.FacialRecognition import FacialRecognition from deepface.models.FacialRecognition import FacialRecognition

View File

@ -25,8 +25,9 @@ class DlibClient(FacialRecognition):
def forward(self, img: np.ndarray) -> List[float]: def forward(self, img: np.ndarray) -> List[float]:
""" """
Find embeddings with Dlib model Find embeddings with Dlib 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: Args:
img (np.ndarray): pre-loaded image in BGR img (np.ndarray): pre-loaded image in BGR
Returns Returns

View File

@ -1,7 +1,5 @@
from typing import List
import os import os
import gdown import gdown
import numpy as np
from deepface.commons import package_utils, folder_utils from deepface.commons import package_utils, folder_utils
from deepface.commons.logger import Logger from deepface.commons.logger import Logger
from deepface.models.FacialRecognition import FacialRecognition from deepface.models.FacialRecognition import FacialRecognition

View File

@ -1,8 +1,6 @@
from typing import List
import os import os
import zipfile import zipfile
import gdown import gdown
import numpy as np
from deepface.commons import package_utils, folder_utils from deepface.commons import package_utils, folder_utils
from deepface.commons.logger import Logger from deepface.commons.logger import Logger
from deepface.models.FacialRecognition import FacialRecognition from deepface.models.FacialRecognition import FacialRecognition

View File

@ -1,10 +1,8 @@
# built-in dependencies # built-in dependencies
import os import os
from typing import List
# 3rd party dependencies # 3rd party dependencies
import gdown import gdown
import numpy as np
import tensorflow as tf import tensorflow as tf
# project dependencies # project dependencies

View File

@ -1,8 +1,6 @@
from typing import List
import os import os
import gdown import gdown
import tensorflow as tf import tensorflow as tf
import numpy as np
from deepface.commons import package_utils, folder_utils from deepface.commons import package_utils, folder_utils
from deepface.commons.logger import Logger from deepface.commons.logger import Logger
from deepface.models.FacialRecognition import FacialRecognition from deepface.models.FacialRecognition import FacialRecognition

View File

@ -28,7 +28,8 @@ class SFaceClient(FacialRecognition):
def forward(self, img: np.ndarray) -> List[float]: def forward(self, img: np.ndarray) -> List[float]:
""" """
Find embeddings with SFace model 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: Args:
img (np.ndarray): pre-loaded image in BGR img (np.ndarray): pre-loaded image in BGR
Returns Returns