pull/136/head
Shawn-Shan 2021-05-10 21:56:41 -05:00
rodzic 5d1c2ad2d7
commit 65743ef509
1 zmienionych plików z 2 dodań i 0 usunięć

Wyświetl plik

@ -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: