Feb, 2024

IRConStyle:使用对比学习和风格迁移的图像修复框架

TL;DR通过对比学习范式在低级任务上应用(如图像修复)的局限性及有效性的不确定性进行深入分析,提出了三个解决问题的准则,并结合对比学习的灵活性,基于风格转换提出了一种用于图像修复的新模块 ConStyle,并将其高效集成到任何 U-Net 结构的网络中,形成一个图像修复框架 (IRConStyle),通过在各种图像修复任务上的广泛实验,包括去噪、去模糊、去雨和去雾,验证了 ConStyle 的能力和兼容性,通过替换通用修复网络为基于 transformer、CNN 和 MLP 的网络,ConStyle NAFNet 在 SOTS 户外(去雾)和 Rain100H(去雨)数据集上分别提升了 4.16 dB 和 3.58 dB,并且参数数量减少了 85%。