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_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]