From 910f323c5fb6a2f3b5875dee60070fb81bcd85ec Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Wed, 14 Aug 2019 11:32:01 -0700 Subject: [PATCH] remove debug strings --- models/Model_SAE/Model.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index eb6ef79..9c65daa 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -575,8 +575,6 @@ class SAEModel(ModelBase): def predictor_func(self, face): if self.options['learn_mask']: - print('np.shape(face):', np.shape(face)) - print('np.shape(face[np.newaxis, ...]):', np.shape(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] return bgr[0], mask[..., 0]