github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-15 01:23:44 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix for amd

Browse source
This commit is contained in:
Colombo 2019-10-24 21:59:13 +04:00
parent 0448a461f5
commit 75eeef0a96
2 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
facelib/S3FDExtractor.py
View file

@ -7,7 +7,7 @@ import numpy as np
from nnlib import nnlib
class S3FDExtractor(object):
def __init__(self):
def __init__(self, do_dummy_predict=False):
exec( nnlib.import_all(), locals(), globals() )
model_path = Path(__file__).parent / "S3FD.h5"
@ -16,7 +16,8 @@ class S3FDExtractor(object):
self.model = nnlib.keras.models.load_model ( str(model_path) )
self.extract ( np.zeros( (1080,1920,3), dtype=np.uint8) )
if do_dummy_predict:
self.extract ( np.zeros( (640,640,3), dtype=np.uint8) )
def __enter__(self):
return self