3 个月前

基于多步强化学习的单图像超分辨率

基于多步强化学习的单图像超分辨率

摘要

深度学习(Deep Learning, DL)因其强大的表达能力和广泛的适用性,已成为当前图像处理研究中的主流技术。在单图像超分辨率(Single Image Super-Resolution, SISR)领域,深度学习凭借从低分辨率(Low-Resolution, LR)图像重建高分辨率(High-Resolution, HR)图像的能力,占据了主导地位,尤其得益于生成对抗网络(Generative Adversarial Networks, GANs)的广泛应用。SISR研究受到广泛关注,主要因其在提升后续图像处理任务性能方面的潜力,例如目标检测、定位与分类等。本研究提出一种基于多智能体强化学习(Multi-Agent Reinforcement Learning, RL)的算法应用于SISR,构建了一种先进的集成方法,用于融合多个性能优异的GAN模型。在该实现中,每个智能体从一个固定的动作集合中选择特定动作,该动作集由现有GAN-SISR算法的输出结果构成,用于更新其对应像素的值。通过像素级或图像块级(patch-wise)的智能体布局以及相应的奖励机制,算法能够学习到一种优化策略,即从各个候选方案中选取最优的像素值,从而有效提升图像的分辨率。

基准测试

基准方法指标
image-super-resolution-on-div2k-val-4xPixelRL-SR
PSNR: 28.08
SSIM: 0.8140
image-super-resolution-on-urban100-4xPixelRL-SR
PSNR: 23.28
SSIM: 0.7517

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于多步强化学习的单图像超分辨率 | 论文 | HyperAI超神经