4 个月前

Pix2Next:利用视觉基础模型实现RGB到NIR图像转换

Pix2Next:利用视觉基础模型实现RGB到NIR图像转换

摘要

本文提出了一种名为Pix2Next的新颖图像到图像转换框架,旨在解决从RGB输入生成高质量近红外(NIR)图像的挑战。我们的方法在编码器-解码器架构中利用了最先进的视觉基础模型(VFM),并通过引入交叉注意力机制来增强特征融合。该设计不仅捕捉到了详细的全局表示,还保留了关键的光谱特性,将RGB到NIR的转换视为不仅仅是简单的域迁移问题。多尺度PatchGAN判别器确保了在不同细节层次上生成逼真的图像,而精心设计的损失函数则将全局上下文理解与局部特征保留相结合。我们在RANUS数据集上进行了实验,以展示Pix2Next在定量指标和视觉质量方面的优势,相比现有方法,FID分数提高了34.81%。此外,我们通过使用生成的NIR数据扩充有限的真实NIR数据集,展示了Pix2Next在下游目标检测任务中的实际应用价值,性能得到了显著提升。所提出的这种方法能够在不增加额外数据采集或标注工作的情况下扩大NIR数据集规模,有望加速基于NIR的计算机视觉应用的发展。

代码仓库

Yonsei-STL/pix2next
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
image-to-image-translation-on-flirPix2Next
PSNR: 23.45
SSIM: 0.66

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
Pix2Next:利用视觉基础模型实现RGB到NIR图像转换 | 论文 | HyperAI超神经