fix s3fd extractor bug for 11GB+ cards

2019-10-17 22:25:23 +04:00
parent e3f63a7b40
commit e63e89c305
2 changed files with 2 additions and 0 deletions
--- a/facelib/S3FDExtractor.py
+++ b/facelib/S3FDExtractor.py
@@ -15,6 +15,8 @@ class S3FDExtractor(object):
            return None

        self.model = nnlib.keras.models.load_model ( str(model_path) )
+        
+        self.extract ( np.zeros( (1080,1920,3), dtype=np.uint8) )

    def __enter__(self):
        return self
--- a/models/archived_models.zip
+++ b/models/archived_models.zip