From 65743ef509beb07b7bf52a7a7af3f4e10ed67e6a Mon Sep 17 00:00:00 2001 From: Shawn-Shan Date: Mon, 10 May 2021 21:56:41 -0500 Subject: [PATCH] minor fix --- fawkes/align_face.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/fawkes/align_face.py b/fawkes/align_face.py index 92f7e9a..d6b5bb5 100644 --- a/fawkes/align_face.py +++ b/fawkes/align_face.py @@ -29,12 +29,14 @@ def align(orig_img, aligner): if dic['confidence'] < 0.9: continue x, y, width, height = dic['box'] + if width < 30 or height < 30: continue bb = [y, x, y + height, x + width] cropped = orig_img[bb[0]:bb[2], bb[1]:bb[3], :] cropped_arr.append(np.copy(cropped)) bounding_boxes_arr.append(bb) + return cropped_arr, bounding_boxes_arr # if nrof_faces > 0: