github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 03:55: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) Four-dimensional numpy array (n, 224, 224, 3)
""" """
image_batch = np.array(img) image_batch = np.array(img)
# Remove batch dimension in advance if exists
image_batch = image_batch.squeeze()
# 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