From ca815abaf5a260de1306e18856abcef05fc7cf78 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Wed, 14 Aug 2019 10:23:52 -0700 Subject: [PATCH] debug predictor --- models/Model_SAE/Model.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index 6330cbd..b70ea9e 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -575,6 +575,10 @@ class SAEModel(ModelBase): def predictor_func(self, face): if self.options['learn_mask']: + print('np.shape(face):', np.shape(face)) + print('np.newaxis:', 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, ...]]) mask = mask_dst_dstm[0] * mask_src_dstm[0] return bgr[0], mask[..., 0]