From 85e2d8d863abf11d2d3cb3bb98c1702857a7aeb2 Mon Sep 17 00:00:00 2001 From: NatLee Date: Tue, 7 Jan 2025 04:05:11 +0800 Subject: [PATCH] [update] modify comment for multi models --- 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 64e56e2..d0a00f1 100644 --- a/deepface/models/Demography.py +++ b/deepface/models/Demography.py @@ -28,7 +28,7 @@ class Demography(ABC): And switch to batch prediction if receives batched images. Args: - img_batch: Batch of images as np.ndarray (n, 224, 224, 3), with n >= 1. + img_batch: Batch of images as np.ndarray (n, x, y, c), with n >= 1, x = image width, y = image height, c = channel """ if not self.model_name: # Check if called from derived class raise NotImplementedError("virtual method must not be called directly")