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

linting updates

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-10-06 10:18:58 +01:00
parent ee4f8e6f02
commit 6d71177ff2
3 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
deepface/models/face_detection/YuNet.py
View File

@ -13,6 +13,7 @@ from deepface.commons.logger import Logger
logger = Logger() logger = Logger()
# pylint:disable=line-too-long
WEIGHTS_URL = "https://github.com/opencv/opencv_zoo/raw/main/models/face_detection_yunet/face_detection_yunet_2023mar.onnx" WEIGHTS_URL = "https://github.com/opencv/opencv_zoo/raw/main/models/face_detection_yunet/face_detection_yunet_2023mar.onnx"

17
deepface/models/facial_recognition/Facenet.py
View File

@ -39,8 +39,13 @@ else:
from tensorflow.keras.layers import add from tensorflow.keras.layers import add
from tensorflow.keras import backend as K from tensorflow.keras import backend as K
FACENET128_WEIGHTS="https://github.com/serengil/deepface_models/releases/download/v1.0/facenet_weights.h5" # pylint:disable=line-too-long
FACENET512_WEIGHTS="https://github.com/serengil/deepface_models/releases/download/v1.0/facenet512_weights.h5" FACENET128_WEIGHTS = (
"https://github.com/serengil/deepface_models/releases/download/v1.0/facenet_weights.h5"
)
FACENET512_WEIGHTS = (
"https://github.com/serengil/deepface_models/releases/download/v1.0/facenet512_weights.h5"
)
# -------------------------------- # --------------------------------
@ -1671,9 +1676,7 @@ def load_facenet128d_model(
weight_file = weight_utils.download_weights_if_necessary( weight_file = weight_utils.download_weights_if_necessary(
file_name="facenet_weights.h5", source_url=url file_name="facenet_weights.h5", source_url=url
) )
model = weight_utils.load_model_weights( model = weight_utils.load_model_weights(model=model, weight_file=weight_file)
model=model, weight_file=weight_file
)
return model return model
@ -1692,8 +1695,6 @@ def load_facenet512d_model(
weight_file = weight_utils.download_weights_if_necessary( weight_file = weight_utils.download_weights_if_necessary(
file_name="facenet512_weights.h5", source_url=url file_name="facenet512_weights.h5", source_url=url
) )
model = weight_utils.load_model_weights( model = weight_utils.load_model_weights(model=model, weight_file=weight_file)
model=model, weight_file=weight_file
)
return model return model

3
deepface/models/facial_recognition/FbDeepFace.py
View File

@ -30,10 +30,9 @@ else:
Dropout, Dropout,
) )
# pylint: disable=line-too-long, too-few-public-methods
WEIGHTS_URL="https://github.com/swghosh/DeepFace/releases/download/weights-vggface2-2d-aligned/VGGFace2_DeepFace_weights_val-0.9034.h5.zip" WEIGHTS_URL="https://github.com/swghosh/DeepFace/releases/download/weights-vggface2-2d-aligned/VGGFace2_DeepFace_weights_val-0.9034.h5.zip"
# -------------------------------------
# pylint: disable=line-too-long, too-few-public-methods
class DeepFaceClient(FacialRecognition): class DeepFaceClient(FacialRecognition):
""" """
Fb's DeepFace model class Fb's DeepFace model class