CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
iperov
2019-08-25 07:43:22 +04:00
parent b572b8f4d4
commit 00dce38187
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models/Model_DF/Model.py
View File

@@ -113,7 +113,7 @@ class Model(ModelBase):
def predictor_func (self, face=None, dummy_predict=False):
if dummy_predict:
self.AE_convert ([ np.zeros ( (1, 128, 128, 3) ), dtype=np.float32 ) ])
self.convert ([ np.zeros ( (1, 128, 128, 3), dtype=np.float32 ) ])
else:
x, mx = self.convert ( [ face[np.newaxis,...] ] )
return x[0], mx[0][...,0]

2
models/Model_H128/Model.py
View File

@@ -121,7 +121,7 @@ class Model(ModelBase):
def predictor_func (self, face=None, dummy_predict=False):
if dummy_predict:
self.AE_convert ([ np.zeros ( (1, 128, 128, 3) ), dtype=np.float32 ) ])
self.src_view ([ np.zeros ( (1, 128, 128, 3), dtype=np.float32 ) ])
else:
x, mx = self.src_view ( [ face[np.newaxis,...] ] )
return x[0], mx[0][...,0]

2
models/Model_H64/Model.py
View File

@@ -122,7 +122,7 @@ class Model(ModelBase):
def predictor_func (self, face=None, dummy_predict=False):
if dummy_predict:
self.AE_convert ([ np.zeros ( (1, 64, 64, 3) ), dtype=np.float32 ) ])
self.src_view ([ np.zeros ( (1, 64, 64, 3), dtype=np.float32 ) ])
else:
x, mx = self.src_view ( [ face[np.newaxis,...] ] )
return x[0], mx[0][...,0]

2
models/Model_LIAEF128/Model.py
View File

@@ -119,7 +119,7 @@ class Model(ModelBase):
def predictor_func (self, face=None, dummy_predict=False):
if dummy_predict:
self.AE_convert ([ np.zeros ( (1, 128, 128, 3) ), dtype=np.float32 ) ])
self.convert ([ np.zeros ( (1, 128, 128, 3), dtype=np.float32 ) ])
else:
x, mx = self.convert ( [ face[np.newaxis,...] ] )
return x[0], mx[0][...,0]