Uformer: A General U-Shaped Transformer for Image Restoration

Understand

In this paper, we present Uformer, an effective and efficient Transformer-based architecture for image restoration, in which we build a hierarchical encoder-decoder network using the Transformer block.

• In Uformer, there are two core designs.
• First, we introduce a novel locally-enhanced window (LeWin) Transformer block, which performs nonoverlapping window-based self-attention instead of global self-attention.
• It significantly reduces the computational complexity on high resolution feature map while capturing local context.