zhengchen1999 / DAT

PyTorch code for our ICCV 2023 paper "Dual Aggregation Transformer for Image Super-Resolution"
Apache License 2.0
350 stars 27 forks source link

DAT model for other restoration work #12

Closed silvadreamer closed 8 months ago

silvadreamer commented 9 months ago

Can DAT model works well on other restoration works?Like deblur or denoise?

zhengchen1999 commented 9 months ago

Hi, Since the architecture of DAT is similar to SwinIR, it can be applied to other restoration works by replacing the reconstruction module. For more information, you can refer to SwinIR. But since DAT is designed for SR, the performance on other tasks may be not very good.

You can also apply the DSTB and DCTB to UNet architecture, which I think should be better than the current architecture of DAT in deblur or denoise tasks.

silvadreamer commented 9 months ago

Hi, Since the architecture of DAT is similar to SwinIR, it can be applied to other restoration works by replacing the reconstruction module. For more information, you can refer to SwinIR. But since DAT is designed for SR, the performance on other tasks may be not very good.

You can also apply the DSTB and DCTB to UNet architecture, which I think should be better than the current architecture of DAT in deblur or denoise tasks.

Thanks a lot. It's a great work.