diff --git a/modules/deeplearning/esrgan_exp.py b/modules/deeplearning/esrgan_exp.py index a53675d5e038f3fbf0df700e2f047f5e48940723..c068ca868fc25056080727e67376c8b31e58e1d9 100644 --- a/modules/deeplearning/esrgan_exp.py +++ b/modules/deeplearning/esrgan_exp.py @@ -21,10 +21,10 @@ model = hub.load(SAVED_MODEL_PATH) def get_image(in_file): h5f = h5py.File(in_file, 'r') - # s_x = slice(2622, 3134) - # s_y = slice(2622, 3134) - s_x = slice(1854, 3902) - s_y = slice(1854, 3902) + s_x = slice(2622, 3134) + s_y = slice(2622, 3134) + # s_x = slice(1854, 3902) + # s_y = slice(1854, 3902) # bt = get_grid_values_all(h5f, 'temp_11_0um_nom') # bt = bt[s_y, s_x] @@ -78,6 +78,7 @@ def run(in_file): # model = hub.load(SAVED_MODEL_PATH) hr_image = get_image(in_file) hr_image = preprocess_image(hr_image) + print('start inference: ', hr_image.shape) t0 = time.time() fake_image = model(hr_image) t1 = time.time()