github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-05 19:15:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

added review changes

Browse Source
This commit is contained in:
Raghucharan16 2025-03-01 14:01:59 +05:30
parent 112eb0a6c2
commit 91319705b1
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
deepface/models/facial_recognition/Buffalo_L.py
View File

@ -25,7 +25,8 @@ class Buffalo_L(FacialRecognition):
raise ModuleNotFoundError(
"InsightFace and its dependencies are optional for the Buffalo_L model. "
"Please install them with: "
"pip install insightface>=0.7.3 onnxruntime>=1.9.0 typing-extensions pydantic albumentations"
"pip install insightface>=0.7.3 onnxruntime>=1.9.0 typing-extensions pydantic"
"albumentations"
) from err
# Define the model filename and subdirectory
@ -86,7 +87,8 @@ class Buffalo_L(FacialRecognition):
"""
Extract face embedding from a pre-cropped face image.
Args:
img: Preprocessed face image with shape (1, 112, 112, 3) or batch (batch_size, 112, 112, 3)
img: Preprocessed face image with shape (1, 112, 112, 3)
or batch (batch_size, 112, 112, 3)
Returns:
Face embedding as a list of floats or list of lists of floats
"""
@ -98,4 +100,5 @@ class Buffalo_L(FacialRecognition):
embedding = np.array(embedding).flatten()
else:
raise ValueError(f"Unexpected embedding type: {type(embedding)}")
return embedding.tolist()
return embedding.tolist()