Merge pull request #115 from mowshon/patch-1
Face detection checkpoint will be downloaded automatically.
This commit is contained in:
commit
d41b9b1020
1 changed files with 1 additions and 1 deletions
|
|
@ -18,7 +18,7 @@ class SFDDetector(FaceDetector):
|
|||
super(SFDDetector, self).__init__(device, verbose)
|
||||
|
||||
# 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'])
|
||||
else:
|
||||
model_weights = torch.load(path_to_detector)
|
||||
|
|
|
|||
Loading…
Reference in a new issue