github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

S3FD and 2DFAN-4 were improperly ported from pytorch. now fixed.

Browse source
This commit is contained in:
Colombo 2019-09-20 17:16:37 +04:00
parent e928ee0d30
commit d9d10f91c2
5 changed files with 98 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

2
mainscripts/Extractor.py
View file

@ -85,7 +85,7 @@ class ExtractSubprocessor(Subprocessor):
elif self.type == 'landmarks':
nnlib.import_all (device_config)
self.e = facelib.LandmarksExtractor(nnlib.keras)
self.e = facelib.FANExtractor()
self.e.__enter__()
if self.device_vram >= 2:
self.second_pass_e = facelib.S3FDExtractor()