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

Added user control for smoothing face dets

Browse source
This commit is contained in:
prajwalkr 2020-09-13 08:57:51 +05:30
parent 0600d0f4da
commit 8549b79e14
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
inference.py
View file

@ -46,6 +46,9 @@ parser.add_argument('--rotate', default=False, action='store_true',
help='Sometimes videos taken from a phone can be flipped 90deg. If true, will flip video right by 90deg.'
'Use if you get a flipped result, despite feeding a normal looking video')
parser.add_argument('--nosmooth', default=False, action='store_true',
help='Prevent smoothing face detections over a short temporal window')
args = parser.parse_args()
args.img_size = 96
@ -94,7 +97,7 @@ def face_detect(images):
results.append([x1, y1, x2, y2])
boxes = get_smoothened_boxes(np.array(results), T=5)
if not args.nosmooth: boxes = get_smoothened_boxes(np.array(results), T=5)
results = [[image[y1: y2, x1:x2], (y1, y2, x1, x2)] for image, (x1, y1, x2, y2) in zip(images, boxes)]
del detector