va1shn9v / PromptIR

PromptIR: Prompting for All-in-One Blind Image Restoration [NeurIPS 2023]
https://arxiv.org/abs/2306.13090
Other
346 stars 26 forks source link

Question about Possible Mismatch between Feature Size and Prompt Size #21

Closed csguoh closed 11 months ago

csguoh commented 11 months ago

Hi, Thanks for this incredible work! I have a question about the possible different feature size between pre-defined prompts and image features. It seems that the spatial size of prompts are pre-defined (such as 64x64 ) and fixed, so if the feature size of one testing image does not match this pre-defined size, hot to handle this? Thx.

va1shn9v commented 11 months ago

Hi, thanks for your interest in our work, we just upsample the prompt Tensors, to match the size/resolution of the features.

csguoh commented 11 months ago

Got it! Thanks for your reply :D