JingyunLiang / SwinIR

SwinIR: Image Restoration Using Swin Transformer (official repository)

Home Page:https://arxiv.org/abs/2108.10257

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to determine patch and window sizes?

subiawaud opened this issue · comments

Hi and thanks for your great work,

How would I go about determining compatible window and patch sizes for a custom dataset, say of size (750,450)?

Just use original window and patch sizes. You can pad the image as a multiple of window size (given window size (8,8), pad it to (752, 456)). In fact, the current code will do it before testing.