github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 05:22:06 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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): def predictor_func (self, face=None, dummy_predict=False):
if dummy_predict: 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: else:
x, mx = self.convert ( [ face[np.newaxis,...] ] ) x, mx = self.convert ( [ face[np.newaxis,...] ] )
return x[0], mx[0][...,0] 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): def predictor_func (self, face=None, dummy_predict=False):
if dummy_predict: 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: else:
x, mx = self.src_view ( [ face[np.newaxis,...] ] ) x, mx = self.src_view ( [ face[np.newaxis,...] ] )
return x[0], mx[0][...,0] 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): def predictor_func (self, face=None, dummy_predict=False):
if dummy_predict: 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: else:
x, mx = self.src_view ( [ face[np.newaxis,...] ] ) x, mx = self.src_view ( [ face[np.newaxis,...] ] )
return x[0], mx[0][...,0] 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): def predictor_func (self, face=None, dummy_predict=False):
if dummy_predict: 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: else:
x, mx = self.convert ( [ face[np.newaxis,...] ] ) x, mx = self.convert ( [ face[np.newaxis,...] ] )
return x[0], mx[0][...,0] return x[0], mx[0][...,0]