SwinIR图片修复工具

联合创作 · 2023-09-29 23:09

这个资源库是SwinIR的官方PyTorch实现:使用移窗变换器修复图像(arxiv,supp,预训练模型,视觉结果)。SwinIR 在以下方面实现了最先进的性能。

  • 二次元/轻量级/真实世界的图像SR
  • 灰度/彩色图像去噪
  • 减少 JPEG 压缩伪影
Real-World Image (x4) BSRGAN, ICCV2021 Real-ESRGAN SwinIR (ours) SwinIR-Large (ours)

训练

使用过的训练和测试集可以下载如下:

任务 训练集 测试集 视觉结果
经典/轻量级图像 SR DIV2K(800 张训练图像)或 DIV2K + Flickr2K(2650 张图像) Set5 + Set14 + BSD100 + Urban100 + Manga109 全部下载 这里
真实世界图像SR SwinIR-M(中等尺寸):DIV2K(800 张训练图像)+ Flickr2K(2650 张图像)+ OST替代链接,10324 张天空、水、草、山、建筑、植物、动物图像)
SwinIR-L(大尺寸) :DIV2K + Flickr2K + OST + WED(4744 张图像)+ FFHQ(前 2000 张图像,人脸)+ Manga109(漫画)+ SCUT-CTW1500(前 100 张训练图像,文本)
RealSRSet+5images 这里
彩色/灰度图像去噪 DIV2K(800 张训练图像)+ Flickr2K(2650 张图像)+ BSD500(400 张训练和测试图像)+ WED(4744 张图像) 灰度:Set12 + BSD68 + Urban100
颜色:CBSD68 + Kodak24 + McMaster + Urban100 下载全部
这里
JPEG 压缩伪影减少 DIV2K(800 张训练图像)+ Flickr2K(2650 张图像)+ BSD500(400 张训练和测试图像)+ WED(4744 张图像) 灰度:Classic5+LIVE1 下载全部 这里
浏览 4
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报