github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-06 11:35:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fix] handle between grayscale and RGB image for models

Browse Source
This commit is contained in:
NatLee 2025-01-07 04:55:14 +08:00
parent 29141b3cd5
commit 36fb512bec
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
deepface/models/Demography.py
View File

@ -36,7 +36,8 @@ class Demography(ABC):
assert img_batch.ndim == 4, "expected 4-dimensional tensor input" assert img_batch.ndim == 4, "expected 4-dimensional tensor input"
if img_batch.shape[0] == 1: # Single image if img_batch.shape[0] == 1: # Single image
img_batch = img_batch.squeeze(0) # Remove batch dimension if img_batch.shape[-1] != 3: # Check if grayscale
img_batch = img_batch.squeeze(0) # Remove batch dimension
predict_result = self.model(img_batch, training=False).numpy()[0, :] # Predict with legacy method. predict_result = self.model(img_batch, training=False).numpy()[0, :] # Predict with legacy method.
return predict_result return predict_result
else: # Batch of images else: # Batch of images