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

didn't make a difference

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-14 11:52:14 -07:00
commit a93cd32337
1 changed files with 3 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

12
models/Model_SAE/Model.py
View file

@ -433,19 +433,13 @@ class SAEModel(ModelBase):
self.AE_view = K.function([warped_src, warped_dst], self.AE_view = K.function([warped_src, warped_dst],
[pred_src_src[-1], pred_dst_dst[-1], pred_src_dst[-1]]) [pred_src_src[-1], pred_dst_dst[-1], pred_src_dst[-1]])
else: else:
# Updated "pred_src_dst[-1]" to "pred_src_dst[-1][:4]", as it was breaking on images with 4th channel # Updated "pred_src_dst[-1]" to "pred_src_dst[-1][:4]", as it was breaking on images with 4th channel
if self.options['learn_mask']: if self.options['learn_mask']:
if self.options['apply_random_ct']: self.AE_convert = K.function([warped_dst], [pred_src_dst[-1][:4], pred_dst_dstm[-1], pred_src_dstm[-1]])
params = [pred_src_dst[-1], pred_dst_dstm[-1], pred_src_dstm[-1]]
else: else:
params = [pred_src_dst[-1][:4], pred_dst_dstm[-1], pred_src_dstm[-1]] self.AE_convert = K.function([warped_dst], [pred_src_dst[-1][:4]])
else:
if self.options['apply_random_ct']:
params = [pred_src_dst[-1]]
else:
params = [pred_src_dst[-1][:4]]
self.AE_convert = K.function([warped_dst], params)
if self.is_training_mode: if self.is_training_mode:
self.src_sample_losses = [] self.src_sample_losses = []