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: Image batch dimension not expanded.

Browse Source
This commit is contained in:
h-alice 2025-01-03 16:22:55 +08:00
parent bb820a7ef4
commit e1822851a5
No known key found for this signature in database
GPG Key ID: 5708F34144A70909
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deepface/models/Demography.py
View File

@ -43,6 +43,6 @@ class Demography(ABC):
# Check input dimension # Check input dimension
if len(image_batch.shape) == 3: if len(image_batch.shape) == 3:
# Single image - add batch dimension # Single image - add batch dimension
imgs = np.expand_dims(image_batch, axis=0) image_batch = np.expand_dims(image_batch, axis=0)
return image_batch return image_batch