github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 12:05:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

raising exceptions for model weights and 3D img checks

Browse Source
This commit is contained in:
raghucharan16 2025-02-27 11:55:35 +05:30
parent 655a221650
commit 0d7c36a63b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deepface/models/facial_recognition/Buffalo_L.py
View File

@ -50,7 +50,7 @@ class Buffalo_L(FacialRecognition):
if os.path.exists(weights_path): if os.path.exists(weights_path):
logger.debug(f"Model file found at: {weights_path}") logger.debug(f"Model file found at: {weights_path}")
else: else:
logger.debug(f"Model file NOT found at: {weights_path}") raise FileNotFoundError(f"Model file not found at: {weights_path}")
# Load the model using the full absolute path # Load the model using the full absolute path
self.model = get_model(weights_path) self.model = get_model(weights_path)
@ -66,9 +66,11 @@ class Buffalo_L(FacialRecognition):
""" """
if len(img.shape) == 4: if len(img.shape) == 4:
img = img[0] img = img[0]
if len(img.shape) != 3:
raise ValueError(f"Expected image to be 3D after preprocessing, but got shape: {img.shape}")
if img.max() <= 1.0: if img.max() <= 1.0:
img = (img * 255).astype(np.uint8) img = (img * 255).astype(np.uint8)
img = img[:, :, ::-1] img = img[:, :, ::-1] # Convert RGB to BGR
return img return img
def forward(self, img: np.ndarray) -> List[float]: def forward(self, img: np.ndarray) -> List[float]: