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

Add comment.

Browse Source
This commit is contained in:
h-alice 2025-01-06 11:51:22 +08:00
parent 5747d9648b
commit 5a1881492f
No known key found for this signature in database
GPG Key ID: 5708F34144A70909
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
deepface/models/Demography.py
View File

@ -37,8 +37,7 @@ class Demography(ABC):
if img_batch.shape[0] == 1: # Single image
img_batch = img_batch.squeeze(0) # Remove batch dimension
predict_result = self.model(img_batch, training=False).numpy()[0, :]
predict_result = np.expand_dims(predict_result, axis=0) # Add batch dimension
predict_result = self.model(img_batch, training=False).numpy()[0, :] # Predict with legacy method.
return predict_result
else: # Batch of images
return self.model.predict_on_batch(img_batch)