diff --git a/modules/deeplearning/esrgan_exp.py b/modules/deeplearning/esrgan_exp.py index 096d237392085e53df5fb54e7739d759acb480da..007874e57b9cec25e13baeb292a02be23df3ab93 100644 --- a/modules/deeplearning/esrgan_exp.py +++ b/modules/deeplearning/esrgan_exp.py @@ -23,8 +23,8 @@ def get_image(in_file): # s_x = slice(2622, 3134) # s_y = slice(2622, 3134) - s_x = slice(2366, 3390) - s_y = slice(2366, 3390) + s_x = slice(2238, 3518) + s_y = slice(2238, 3518) # s_x = slice(1854, 3902) # s_y = slice(1854, 3902) @@ -78,8 +78,11 @@ def preprocess_image(hr_image): def run(in_file): # model = hub.load(SAVED_MODEL_PATH) + t0 = time.time() hr_image = get_image(in_file) hr_image = preprocess_image(hr_image) + t1 = time.time() + print('processing time: ', (t1-t0)) print('start inference: ', hr_image.shape) t0 = time.time() fake_image = model(hr_image)