From e1822851a56dc9a8e5cf7e8264ad2b9e2ed801ee Mon Sep 17 00:00:00 2001 From: h-alice Date: Fri, 3 Jan 2025 16:22:55 +0800 Subject: [PATCH] Fix: Image batch dimension not expanded. --- deepface/models/Demography.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepface/models/Demography.py b/deepface/models/Demography.py index e73fe65..d240f1e 100644 --- a/deepface/models/Demography.py +++ b/deepface/models/Demography.py @@ -43,6 +43,6 @@ class Demography(ABC): # Check input dimension if len(image_batch.shape) == 3: # 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