diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index 54e15bf5214f9ba9fa8aa73d18d1ee62bfee8458..d27784dec410d221dfcbf0a261023ca025af45b4 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -689,7 +689,7 @@ class ESPCN: ckpt.restore(ckpt_manager.latest_checkpoint) - data = normalize(nda_lr, data_param, mean_std_dct) + data = normalize(nda_lr, param, mean_std_dct) data = np.expand_dims(data, axis=0) data = np.expand_dims(data, axis=3)