diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index 764af4b1c7fbcc70280843ced70ac9a9276a6c3d..6ddb3f14d8b04f9980751a5bbad5d2a1b25299c0 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -817,7 +817,8 @@ class ESPCN: def prepare(param_idx=1, filename='/Users/tomrink/data_valid_40.npy'): nda = np.load(filename) - nda = nda[:, param_idx, :, :] + #nda = nda[:, param_idx, :, :] + nda = nda[:, param_idx, 3:67, 3:67] nda = np.expand_dims(nda, axis=3) # nda_lr = tf.image.resize(nda, (36, 36)).numpy() nda_lr = tf.image.resize(nda, (32, 32)).numpy() @@ -826,7 +827,6 @@ def prepare(param_idx=1, filename='/Users/tomrink/data_valid_40.npy'): def run_evaluate_static(nda_lr, param='temp_11_0um_nom', ckpt_dir='/Users/tomrink/tf_model_sres/run-20220805173619/'): nn = ESPCN() - # out_sr = nn.run_evaluate(nda_lr[2, :, :, 0], param, ckpt_dir) out_sr = nn.run_evaluate(nda_lr, param, ckpt_dir) return out_sr