From 801e2f4beac04906072443cc6196b923d9726600 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Wed, 14 Aug 2019 10:40:24 -0700 Subject: [PATCH] debugging --- models/Model_SAE/Model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index bfdd4c2..9a09c65 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -576,6 +576,7 @@ 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, ...])) 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, ...]])