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

[fix] check for input number of faces

Browse Source
This commit is contained in:
NatLee 2025-01-07 05:26:27 +08:00
parent 041773232f
commit c44af00269
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
deepface/modules/demography.py
View File

@ -207,12 +207,14 @@ def analyze(
age_predictions = model.predict(faces_array) age_predictions = model.predict(faces_array)
# Handle single vs multiple age predictions # Handle single vs multiple age predictions
if len(age_predictions.shape) == 1: if faces_array.shape[0] == 1:
# Single face case - reshape predictions to 2D array for consistent handling # Single face case - reshape predictions to 2D array for consistent handling
age_predictions = age_predictions.reshape(1, -1) resp_objects[idx]["age"] = int(np.argmax(age_predictions))
else:
# Multiple face case - iterate over each prediction
for idx, age in enumerate(age_predictions): for idx, age in enumerate(age_predictions):
resp_objects[idx]["age"] = np.argmax(age) resp_objects[idx]["age"] = int(age)
elif action == "gender": elif action == "gender":
# Build the gender model # Build the gender model