arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2208.01837cs.CV

学习先验特征与注意力增强的图像修复

Learning Prior Feature and Attention Enhanced Image Inpainting

Fudan University(复旦大学)

Chenjie Cao, Qiaole Dong, Yanwei Fu

AI总结:

针对普通CNN骨干网络限制图像修复性能的问题,该研究将预训练MAE引入修复模型,利用其先验特征与注意力增强修复效果,在Places2与FFHQ数据集上验证了有效性并开源相关资源。

摘要:

近年来,许多图像修复工作通过利用深度神经网络(DNNs)建模各类先验信息用于图像恢复,已取得令人瞩目的成果。但遗憾的是,这些方法的性能很大程度上受限于普通卷积神经网络(CNNs)骨干网络的表征能力。另一方面,经过自监督预训练的视觉Transformer(ViT)已在诸多视觉识别和目标检测任务中展现出巨大潜力。一个很自然的问题是:图像修复任务能否从ViT骨干网络中大幅获益?然而,直接在修复网络中替换新型骨干网络并非易事,因为图像修复本质上是一类与识别任务截然不同的逆问题。为此,本文将基于预训练的掩码自编码器(Masked AutoEncoder, MAE)引入图像修复模型,该模型可借助更丰富的信息先验来增强修复过程。此外,我们提出利用来自MAE的注意力先验,使修复模型学习掩码区域与非掩码区域之间更多的长距离依赖关系。本文针对图像修复与自监督预训练模型开展了充分的消融实验讨论。同时,在Places2和FFHQ两个数据集上的实验均验证了我们所提模型的有效性。代码与预训练模型已发布于https://github.com/ewrfcas/MAE-FAR。

英文摘要:

Many recent inpainting works have achieved impressive results by leveraging Deep Neural Networks (DNNs) to model various prior information for image restoration. Unfortunately, the performance of these methods is largely limited by the representation ability of vanilla Convolutional Neural Networks (CNNs) backbones.On the other hand, Vision Transformers (ViT) with self-supervised pre-training have shown great potential for many visual recognition and object detection tasks. A natural question is whether the inpainting task can be greatly benefited from the ViT backbone? However, it is nontrivial to directly replace the new backbones in inpainting networks, as the inpainting is an inverse problem fundamentally different from the recognition tasks. To this end, this paper incorporates the pre-training based Masked AutoEncoder (MAE) into the inpainting model, which enjoys richer informative priors to enhance the inpainting process. Moreover, we propose to use attention priors from MAE to make the inpainting model learn more long-distance dependencies between masked and unmasked regions. Sufficient ablations have been discussed about the inpainting and the self-supervised pre-training models in this paper. Besides, experiments on both Places2 and FFHQ demonstrate the effectiveness of our proposed model. Codes and pre-trained models are released in https://github.com/ewrfcas/MAE-FAR.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑