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
@@ -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