diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index 9d1858d2e24707838adeac6b5df6f30888be7a76..cf02a00032cf429d2298b7117ceb0634f9be3962 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -609,9 +609,9 @@ class SRCNN: ckpt.restore(ckpt_manager.latest_checkpoint) - data = normalize(nda_lr, data_param, mean_std_dct) - data = np.expand_dims(data, axis=0) - data = np.expand_dims(data, axis=3) + data = normalize(nda_lr, param, mean_std_dct) + #data = np.expand_dims(data, axis=0) + #data = np.expand_dims(data, axis=3) self.reset_test_metrics()