Algolzw / BSRT

Pytorch code for "BSRT: Improving Burst Super-Resolution with Swin Transformer and Flow-Guided Deformable Alignment", CVPRW, 1st place in NTIRE 2022 BurstSR Challenge (real-world track).
MIT License
179 stars 13 forks source link

Upscale Factor #11

Closed nonick2k23 closed 1 year ago

nonick2k23 commented 1 year ago

I want to thank you for the assistance at the other thread.

I am wondering if it is possible to modify the network to provide upscale of x2 instead of x8?

I noticed that the network contains three pixel shuffles, each up-scaled by x2, so we have x2 on x2 on x2 -> x8 up-scaled image in the end.

Did you experiment with x2 upscale? Do you have suggestions how to modify the network so x2 upscale works with optimal results?

Thanks

Algolzw commented 1 year ago

Hi, we haven't tried x2 scale upsampling yet. Now our code works on the RAW images so it uses the first x2 to parse 'RGGB' RAW format to a normal image and then uses two x2 to perform x4 SR.

If you want to perform x2 on RAW images you can remove the third x2 operation. And if you want to perform x2 SR on RGB images, you need to remove the first x2 operation as well.

Best.

nonick2k23 commented 1 year ago

Thanks