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: