- ICCV 2021
-本文主要工作:①提出了一种具有有效的局部增强位置编码的条带窗口自注意变压器
②结合全局层和局部层的注意力
③提出了一种新的HSV损失以增强颜色一致性
- 网络结构
如上图所示,给定一张256x256的失真图像和二进制mask并concat,输入到三个下采样的卷积层后,进行通道分割。一部分通道经过上半部分的“全局层”,即CSWIN变压器模块,获取全局的注意力。另一部分通道经过下半部分的“RRDB”,获取局部注意力和残差。而后通过联合注意力将两个通道的特征链接起来,再经过三个上采样层,得到结果。
主要工作:
-Special CSWIN Transformer
- Redesigned CSWIN Block
-Residual Dense Block
- Joint attention