github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-11 07:37:03 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix S3FD for plaidML, fix S3FD warnings on Lambda layers.

Browse source
This commit is contained in:
iperov 2019-03-11 09:32:00 +04:00
parent 338a8fccc6
commit 893354b870
2 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

BIN
facelib/S3FD.h5
View file

Binary file not shown.

2
mainscripts/Extractor.py
View file

@ -213,7 +213,7 @@ class ExtractSubprocessor(Subprocessor):
cpu_only = True cpu_only = True
if not cpu_only and (type == 'rects' or type == 'landmarks'): if not cpu_only and (type == 'rects' or type == 'landmarks'):
if type == 'rects' and (self.detector == 'mt' or self.detector == 's3fd') and nnlib.device.backend == "plaidML": if type == 'rects' and (self.detector == 'mt') and nnlib.device.backend == "plaidML":
cpu_only = True cpu_only = True
else: else:
if multi_gpu: if multi_gpu: