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

Added crop feature

Browse source
This commit is contained in:
prajwalkr 2020-08-30 14:05:21 +05:30
parent 2f51108beb
commit 8c6d1d3947
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
inference.py
View file

@ -34,6 +34,10 @@ parser.add_argument('--wav2lip_batch_size', type=int, help='Batch size for Wav2L
parser.add_argument('--resize_factor', default=1, type=int,
help='Reduce the resolution by this factor. Sometimes, best results are obtained at 480p or 720p')
parser.add_argument('--crop', nargs='+', type=int, default=[0, 0, -1, -1],
help='Crop video to a smaller region (top, bottom, left, right). Applied after resize_factor arg. '
'Useful if multiple face present. -1 implies the value will be auto-inferred based on height, width')
args = parser.parse_args()
args.img_size = 96
@ -180,6 +184,12 @@ def main():
if args.resize_factor > 1:
frame = cv2.resize(frame, (frame.shape[1]//args.resize_factor, frame.shape[0]//args.resize_factor))
y1, y2, x1, x2 = args.crop
if x2 == -1: x2 = frame.shape[1]
if y2 == -1: y2 = frame.shape[0]
frame = frame[y1:y2, x1:x2]
full_frames.append(frame)
print ("Number of frames available for inference: "+str(len(full_frames)))