| import torch | |
| import time | |
| import os | |
| path_id = "" | |
| checkpoint_path="wav2lip/wav2lip_gan.pth" | |
| outfile="out.mp4" | |
| audiofile="tmp.wav" | |
| imgfile="avatar.png" | |
| driverfile="face_vid2vid/assets/driver06.mp4" | |
| animatedfile="animated.mp4" | |
| static=False | |
| fps=25 | |
| pads=[0, 10, 0, 0] | |
| face_det_batch_size=16 | |
| wav2lip_batch_size=128 | |
| resize_factor=0.5 | |
| crop=[0, -1, 0, -1] | |
| box=[-1, -1, -1, -1] | |
| img_size = 96 | |
| rotate=False | |
| nosmooth=False | |
| mel_step_size = 16 | |
| device = 'cuda' if torch.cuda.is_available() else 'cpu' | |
| print('Using {} for inference.'.format(device)) | |
| import warnings | |
| warnings.filterwarnings('ignore') | |
| def init_path_id(): | |
| path_id = str(int(time.time())) | |
| path = os.path.join("temp", path_id) | |
| os.makedirs(path, exist_ok=True) | |
| return path_id, path | |