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

resolving issue 1098

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-03-11 18:39:00 +00:00
parent ec4c3cd17c
commit 820ba5946b
1 changed files with 35 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
deepface/modules/verification.py
View File

@ -93,6 +93,7 @@ def verify(
model: FacialRecognition = modeling.build_model(model_name) model: FacialRecognition = modeling.build_model(model_name)
dims = model.output_shape dims = model.output_shape
# extract faces from img1
if isinstance(img1_path, list): if isinstance(img1_path, list):
# given image is already pre-calculated embedding # given image is already pre-calculated embedding
if not all(isinstance(dim, float) for dim in img1_path): if not all(isinstance(dim, float) for dim in img1_path):
@ -115,16 +116,20 @@ def verify(
img1_embeddings = [img1_path] img1_embeddings = [img1_path]
img1_facial_areas = [None] img1_facial_areas = [None]
else: else:
img1_embeddings, img1_facial_areas = __extract_faces_and_embeddings( try:
img_path=img1_path, img1_embeddings, img1_facial_areas = __extract_faces_and_embeddings(
model_name=model_name, img_path=img1_path,
detector_backend=detector_backend, model_name=model_name,
enforce_detection=enforce_detection, detector_backend=detector_backend,
align=align, enforce_detection=enforce_detection,
expand_percentage=expand_percentage, align=align,
normalization=normalization, expand_percentage=expand_percentage,
) normalization=normalization,
)
except ValueError as err:
raise ValueError("Exception while processing img1_path") from err
# extract faces from img2
if isinstance(img2_path, list): if isinstance(img2_path, list):
# given image is already pre-calculated embedding # given image is already pre-calculated embedding
if not all(isinstance(dim, float) for dim in img2_path): if not all(isinstance(dim, float) for dim in img2_path):
@ -147,15 +152,18 @@ def verify(
img2_embeddings = [img2_path] img2_embeddings = [img2_path]
img2_facial_areas = [None] img2_facial_areas = [None]
else: else:
img2_embeddings, img2_facial_areas = __extract_faces_and_embeddings( try:
img_path=img2_path, img2_embeddings, img2_facial_areas = __extract_faces_and_embeddings(
model_name=model_name, img_path=img2_path,
detector_backend=detector_backend, model_name=model_name,
enforce_detection=enforce_detection, detector_backend=detector_backend,
align=align, enforce_detection=enforce_detection,
expand_percentage=expand_percentage, align=align,
normalization=normalization, expand_percentage=expand_percentage,
) normalization=normalization,
)
except ValueError as err:
raise ValueError("Exception while processing img2_path") from err
no_facial_area = { no_facial_area = {
"x": None, "x": None,
@ -218,18 +226,15 @@ def __extract_faces_and_embeddings(
model: FacialRecognition = modeling.build_model(model_name) model: FacialRecognition = modeling.build_model(model_name)
target_size = model.input_shape target_size = model.input_shape
try: img_objs = detection.extract_faces(
img_objs = detection.extract_faces( img_path=img_path,
img_path=img_path, target_size=target_size,
target_size=target_size, detector_backend=detector_backend,
detector_backend=detector_backend, grayscale=False,
grayscale=False, enforce_detection=enforce_detection,
enforce_detection=enforce_detection, align=align,
align=align, expand_percentage=expand_percentage,
expand_percentage=expand_percentage, )
)
except ValueError as err:
raise ValueError("Exception while processing img1_path") from err
# find embeddings for each face # find embeddings for each face
for img_obj in img_objs: for img_obj in img_objs: