From 95bb92c933dcc2b1c591a6b4de7c992eabb9661a Mon Sep 17 00:00:00 2001 From: h-alice Date: Tue, 21 Jan 2025 11:25:33 +0800 Subject: [PATCH] Remove redundant squeeze. --- deepface/models/Demography.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/deepface/models/Demography.py b/deepface/models/Demography.py index 87869b9..1493059 100644 --- a/deepface/models/Demography.py +++ b/deepface/models/Demography.py @@ -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