Merge pull request #48 from dipam7/master

fix non-smoothed boxes output
This commit is contained in:
Prajwal 2020-09-13 11:27:47 +05:30 committed by GitHub
commit 11a8eac1aa
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -97,7 +97,8 @@ def face_detect(images):
results.append([x1, y1, x2, y2])
if not args.nosmooth: boxes = get_smoothened_boxes(np.array(results), T=5)
boxes = np.array(results)
if not args.nosmooth: boxes = get_smoothened_boxes(boxes, T=5)
results = [[image[y1: y2, x1:x2], (y1, y2, x1, x2)] for image, (x1, y1, x2, y2) in zip(images, boxes)]
del detector