github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-06 11:35:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix batched numpy array input

Browse Source
This commit is contained in:
galthran-wq 2025-02-18 10:27:23 +00:00
parent 526ab1baf7
commit 988afa6a70
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deepface/modules/detection.py
View File

@ -82,6 +82,8 @@ def extract_faces(
just available in the result only if anti_spoofing is set to True in input arguments.
"""
if isinstance(img_path, np.ndarray) and img_path.ndim == 4:
img_path = [img_path[i] for i in range(img_path.shape[0])]
if not isinstance(img_path, list):
img_path = [img_path]