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

Merge pull request #115 from mowshon/patch-1

Browse source 
Face detection checkpoint will be downloaded automatically.
This commit is contained in:
Prajwal 2020-10-22 19:50:51 +05:30 committed by GitHub
commit d41b9b1020
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
face_detection/detection/sfd/sfd_detector.py
View file

@ -18,7 +18,7 @@ class SFDDetector(FaceDetector):
super(SFDDetector, self).__init__(device, verbose) super(SFDDetector, self).__init__(device, verbose)
# Initialise the face detector # Initialise the face detector
if path_to_detector is None: if not os.path.isfile(path_to_detector):
model_weights = load_url(models_urls['s3fd']) model_weights = load_url(models_urls['s3fd'])
else: else:
model_weights = torch.load(path_to_detector) model_weights = torch.load(path_to_detector)