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

2019-09-20 17:16:37 +04:00
parent e928ee0d30
commit d9d10f91c2
5 changed files with 98 additions and 14 deletions
@@ -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()