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 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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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