From 31417bc97ae77e435889e5ad31da8beaa11562e8 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Wed, 14 Aug 2019 12:13:07 -0700 Subject: [PATCH] revert change, found issue in ConverterMasked --- models/Model_SAE/Model.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index f1165f1..6330cbd 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -435,11 +435,10 @@ class SAEModel(ModelBase): else: - # 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']: - self.AE_convert = K.function([warped_dst], [pred_src_dst[-1][:4], pred_dst_dstm[-1], pred_src_dstm[-1]]) + self.AE_convert = K.function([warped_dst], [pred_src_dst[-1], pred_dst_dstm[-1], pred_src_dstm[-1]]) else: - self.AE_convert = K.function([warped_dst], [pred_src_dst[-1][:4]]) + self.AE_convert = K.function([warped_dst], [pred_src_dst[-1]]) if self.is_training_mode: self.src_sample_losses = []