From 301cca6f2bbcb570121c45b5c0cb2a67585bdd03 Mon Sep 17 00:00:00 2001 From: "bogdan.vlad" Date: Thu, 27 Apr 2023 23:20:47 +0300 Subject: [PATCH] do not pad when dimensions of input image is already multiple of window size --- main_test_swin2sr.py | 4 ++-- predict.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/main_test_swin2sr.py b/main_test_swin2sr.py index 9358436..163ae12 100644 --- a/main_test_swin2sr.py +++ b/main_test_swin2sr.py @@ -68,8 +68,8 @@ def main(): with torch.no_grad(): # pad input image to be a multiple of window_size _, _, h_old, w_old = img_lq.size() - h_pad = (h_old // window_size + 1) * window_size - h_old - w_pad = (w_old // window_size + 1) * window_size - w_old + h_pad = (window_size - h_old % window_size) % window_size + w_pad = (window_size - w_old % window_size) % window_size img_lq = torch.cat([img_lq, torch.flip(img_lq, [2])], 2)[:, :, :h_old + h_pad, :] img_lq = torch.cat([img_lq, torch.flip(img_lq, [3])], 3)[:, :, :, :w_old + w_pad] output = test(img_lq, model, args, window_size) diff --git a/predict.py b/predict.py index 32930fc..4cdc461 100644 --- a/predict.py +++ b/predict.py @@ -62,8 +62,8 @@ def predict( with torch.no_grad(): # pad input image to be a multiple of window_size _, _, h_old, w_old = img_lq.size() - h_pad = (h_old // window_size + 1) * window_size - h_old - w_pad = (w_old // window_size + 1) * window_size - w_old + h_pad = (window_size - h_old % window_size) % window_size + w_pad = (window_size - w_old % window_size) % window_size img_lq = torch.cat([img_lq, torch.flip(img_lq, [2])], 2)[ :, :, : h_old + h_pad, : ]