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

don't print this

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-14 10:47:35 -07:00
commit 47c632b87d
1 changed files with 0 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_SAE/Model.py
View file

@ -576,9 +576,7 @@ class SAEModel(ModelBase):
def predictor_func(self, face): def predictor_func(self, face):
if self.options['learn_mask']: if self.options['learn_mask']:
print('np.shape(face):', np.shape(face)) print('np.shape(face):', np.shape(face))
print('np.shape(face[np.newaxis, ...]):', np.shape(face[np.newaxis, ...])) print('np.shape(face[np.newaxis, ...]):', np.shape(face[np.newaxis, ...]))
print('np.shape(self.AE_convert([face[np.newaxis, ...]])):', self.AE_convert([face[np.newaxis, ...]]))
bgr, mask_dst_dstm, mask_src_dstm = self.AE_convert([face[np.newaxis, ...]]) bgr, mask_dst_dstm, mask_src_dstm = self.AE_convert([face[np.newaxis, ...]])
mask = mask_dst_dstm[0] * mask_src_dstm[0] mask = mask_dst_dstm[0] * mask_src_dstm[0]
return bgr[0], mask[..., 0] return bgr[0], mask[..., 0]