diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index b7e561c9385fb5d421370652690cded1f9c239c8..36e529040d52357faf211b0a6ae588008481b38f 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -809,6 +809,21 @@ class ESPCN: return self.do_evaluate(nda_lr, param, ckpt_dir) +def prepare(param_idx=1, filename='/Users/tomrink/data_valid_40.npy'): + nda = np.load(filename) + nda = nda[:, param_idx, :, :] + nda = np.expand_dims(nda, axis=3) + nda_lr = tf.image.resize(nda, (32, 32)).numpy() + return nda_lr + + +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 + + if __name__ == "__main__": nn = ESPCN() nn.run('matchup_filename')