patch: fix dimension

serengil · Jan 13, 2025 · 72b6db1 · 72b6db1
1 parent 910d6e1
commit 72b6db1
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/deepface/models/Demography.py b/deepface/models/Demography.py
@@ -40,7 +40,6 @@ def _predict_internal(self, img_batch: np.ndarray) -> np.ndarray:
         assert img_batch.ndim == 4, "expected 4-dimensional tensor input"
 
         if img_batch.shape[0] == 1: # Single image
-            img_batch = img_batch.squeeze(0) # Remove batch dimension
             # Predict with legacy method.
             return self.model(img_batch, training=False).numpy()[0, :]
         else: