From 4e2f8aba88629bde2bf8ec628d033a00a42ebbac Mon Sep 17 00:00:00 2001 From: sotirismos Date: Tue, 18 Jan 2022 22:33:22 +0200 Subject: [PATCH] enhanced cropping of detected faces --- deepface/detectors/DlibWrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepface/detectors/DlibWrapper.py b/deepface/detectors/DlibWrapper.py index 207e816..1a2283d 100644 --- a/deepface/detectors/DlibWrapper.py +++ b/deepface/detectors/DlibWrapper.py @@ -55,7 +55,7 @@ def detect_face(detector, img, align = True): for idx, d in enumerate(detections): left = d.left(); right = d.right() top = d.top(); bottom = d.bottom() - detected_face = img[top:bottom, left:right] + detected_face = img[max(0, top): min(bottom, img.shape[0]), max(0, left): min(right, img.shape[1])] img_region = [left, top, right - left, bottom - top] if align: