if detector is skipped, then resizing is missing in representation

This commit is contained in:
Sefik Ilkin Serengil 2023-02-10 13:11:30 +00:00 committed by GitHub
parent fc3f607bf4
commit ea43de9baa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -639,28 +639,16 @@ def represent(
model = build_model(model_name)
# ---------------------------------
# we have run pre-process in verification. so, this can be skipped if it is coming from verify.
if detector_backend != "skip":
target_size = functions.find_target_size(model_name=model_name)
target_size = functions.find_target_size(model_name=model_name)
img_objs = functions.extract_faces(
img=img_path,
target_size=target_size,
detector_backend=detector_backend,
grayscale=False,
enforce_detection=enforce_detection,
align=align,
)
else: # skip
if isinstance(img_path, str):
img = functions.load_image(img_path)
elif type(img_path).__module__ == np.__name__:
img = img_path.copy()
else:
raise ValueError(f"unexpected type for img_path - {type(img_path)}")
img_region = [0, 0, img.shape[1], img.shape[0]]
img_objs = [(img, img_region, 0)]
img_objs = functions.extract_faces(
img=img_path,
target_size=target_size,
detector_backend=detector_backend,
grayscale=False,
enforce_detection=enforce_detection,
align=align,
)
# ---------------------------------
for img, region, _ in img_objs: