From e2d1d5acb4ec54be384585aac058025ced52a225 Mon Sep 17 00:00:00 2001 From: Raghucharan16 Date: Sat, 1 Mar 2025 14:48:47 +0530 Subject: [PATCH] added review changes --- deepface/models/facial_recognition/Buffalo_L.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/deepface/models/facial_recognition/Buffalo_L.py b/deepface/models/facial_recognition/Buffalo_L.py index 27930ec..eeca234 100644 --- a/deepface/models/facial_recognition/Buffalo_L.py +++ b/deepface/models/facial_recognition/Buffalo_L.py @@ -64,7 +64,7 @@ class Buffalo_L(FacialRecognition): if len(img.shape) == 3: img = np.expand_dims(img, axis=0) # Convert single image to batch of 1 elif len(img.shape) != 4: - raise ValueError(f"Input must have shape (112, 112, 3) or (X, 112, 112, 3). Got {img.shape}") + raise ValueError(f"Input must be (112, 112, 3) or (X, 112, 112, 3). Got {img.shape}") # Convert RGB to BGR for the entire batch img = img[:, :, :, ::-1] return img @@ -92,4 +92,5 @@ class Buffalo_L(FacialRecognition): embeddings.append(embedding.flatten().tolist()) # Return single embedding if batch_size is 1, otherwise return list of embeddings - return embeddings[0] if batch_size == 1 else embeddings \ No newline at end of file + return embeddings[0] if batch_size == 1 else embeddings + \ No newline at end of file