From 1ea31bc0726de1ce0e171eff906ddf3a6bda390c Mon Sep 17 00:00:00 2001 From: dakotah-jones Date: Wed, 15 Jan 2025 16:41:40 -0500 Subject: [PATCH] Update skip case. --- deepface/modules/representation.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/deepface/modules/representation.py b/deepface/modules/representation.py index d880645..d5783eb 100644 --- a/deepface/modules/representation.py +++ b/deepface/modules/representation.py @@ -87,7 +87,10 @@ def represent( else: # skip # Try load. If load error, will raise exception internal img, _ = image_utils.load_image(img_path) - + + # bgr to rgb + img = img[:, :, ::-1] + if len(img.shape) != 3: raise ValueError(f"Input img must be 3 dimensional but it is {img.shape}")