S3FD and 2DFAN-4 were improperly ported from pytorch. now fixed.
This commit is contained in:
1 parent
e928ee0d30
commit
d9d10f91c2
5 files changed
+98
-14
No files matched your search
@@ -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()
|
||||
|
||||
Reference in new issue
Block a user