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

Remove redundant squeeze.

Browse Source
This commit is contained in:
h-alice 2025-01-21 11:25:33 +08:00
parent db4b749c98
commit 95bb92c933
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

@ -62,8 +62,7 @@ class Demography(ABC):
Four-dimensional numpy array (n, 224, 224, 3)
"""
image_batch = np.array(img)
# Remove batch dimension in advance if exists
image_batch = image_batch.squeeze()
# Check input dimension
if len(image_batch.shape) == 3:
# Single image - add batch dimension