pingu_98/Wav2Lip
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix non-smoothed boxes output

Browse source
This commit is contained in:
Dipam Vasani 2020-09-13 01:02:53 -04:00
parent 8549b79e14
commit 47d29fea0d
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
inference.py
View file

@ -97,7 +97,8 @@ def face_detect(images):
results.append([x1, y1, x2, y2]) 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)] results = [[image[y1: y2, x1:x2], (y1, y2, x1, x2)] for image, (x1, y1, x2, y2) in zip(images, boxes)]
del detector del detector