Accurate and rapid image inpainting is of great significance for environmental perception in unmanned surface vehicles (USVs) with soiled lenses. However, existing methods still struggle to handle large-scale missing regions with complex semantics due to the lack of sufficient prior guidance. To address this problem, we propose a novel reference-guided image inpainting network based on progressive reference alignment and fusion for USVs, called RAFNet. Specifically, we devise a new reference alignment (RA) module to locally align the reference features with the corrupted features, thereby generating aligned features to guide image inpainting. Moreover, a cross-stage fusion (CF) module is presented to adaptively fuse multi-scale aligned features from different network stages. Building upon the RA and CF modules, a progressive reference alignment and fusion decoder is designed to gradually aggregate and refine contextual information from both the reference image and the corrupted image, ultimately yielding the inpainted result. Experiments on our maritime stereo image (MSI) dataset show that RAFNet outperforms the state-of-the-art baseline by 13.9% in PSNR and 6.4% in SSIM, while running at over 28 fps on an NVIDIA RTX 4090D GPU.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Reference-Guided Image Inpainting Network for USVs via Progressive Reference Alignment and Fusion

  • Jingyi Liu,
  • Hengyu Li,
  • Hongkun Zhou,
  • Shaorong Xie

摘要

Accurate and rapid image inpainting is of great significance for environmental perception in unmanned surface vehicles (USVs) with soiled lenses. However, existing methods still struggle to handle large-scale missing regions with complex semantics due to the lack of sufficient prior guidance. To address this problem, we propose a novel reference-guided image inpainting network based on progressive reference alignment and fusion for USVs, called RAFNet. Specifically, we devise a new reference alignment (RA) module to locally align the reference features with the corrupted features, thereby generating aligned features to guide image inpainting. Moreover, a cross-stage fusion (CF) module is presented to adaptively fuse multi-scale aligned features from different network stages. Building upon the RA and CF modules, a progressive reference alignment and fusion decoder is designed to gradually aggregate and refine contextual information from both the reference image and the corrupted image, ultimately yielding the inpainted result. Experiments on our maritime stereo image (MSI) dataset show that RAFNet outperforms the state-of-the-art baseline by 13.9% in PSNR and 6.4% in SSIM, while running at over 28 fps on an NVIDIA RTX 4090D GPU.