BasicSR
基于PyTorch的图像与视频复原开源工具箱，集成EDSR/ESRGAN/SwinIR等20+主流算
加载项目详情…
本应用为开源项目，仅供学习研究，请遵守其开源协议。
微信扫码咨询客服
POMOAI — 让开源 AI 即刻运行
基于PyTorch的图像与视频复原开源工具箱，集成EDSR/ESRGAN/SwinIR等20+主流算
加载项目详情…
本应用为开源项目，仅供学习研究，请遵守其开源协议。
BasicSR（Basic Super Restoration）是一款功能强大的开源图像与视频复原工具箱，由 XPixel 团队（隶属于鹏城实验室）开发和维护。项目基于 PyTorch 深度学习框架实现，涵盖了从单图超分辨率到视频复原的完整技术栈，被广泛应用于学术研究和工业部署场景。
想象一下这样的场景：翻出一张十年前的老照片，细节已经模糊得看不清；或者用手机拍摄了一段重要视频，画面却因为抖动和低光照而满是噪点。在过去，这些问题往往只能靠昂贵的专业软件或人工修复。
BasicSR 的出现改变了这一局面。它用深度学习的方式，让计算机学会了"猜"出丢失的细节——给定一张模糊输入，模型能推断出清晰的高分辨率输出。这个过程不是简单的放大（放大只会让模糊更明显），而是通过学习数百万张"低清-高清"图像对，掌握了图像内在的纹理和结构规律。
这个工具箱的命名就透露了它的定位：基础级（Basic）的超分辨率（Super Resolution）研究平台。它不只是某个单一算法的实现，而是集成了近年来该领域几乎所有里程碑式的工作，让研究者和工程师可以在同一个框架下比较、改进和部署这些方法。
BasicSR 最早由开源作者 xinntao（张啸）于 2018 年创建，后来随着 XPixel 团队的加入，项目进入了快速迭代期。2021年，BasicVSR 和 IconVSR 视频超分算法在 CVPR 2021 发表；同年9月加入 SwinIR；2022年5月，项目正式加入 XPixel 团队，进入新的发展阶段。
XPixel 团队（xpixel.group）是图像处理与计算机视觉领域的知名研究团队，隶属于鹏城实验室（PCNL），长期活跃于 CVPR、ICCV、ECCV 等顶级会议，发表了大量图像与视频复原领域的高水平论文。BasicSR 既是他们科研成果的工程化载体，也是服务社区的重要开源基础设施。
图1：BasicSR 项目整体架构
这是 BasicSR 最核心的功能。工具箱实现了以下经典与前沿算法：
对于普通用户来说，最直接的使用方式是：准备一堆低分辨率图片、选一个预训练模型、运行推理脚本、获得高清输出。整个过程不需要写一行代码。
视频超分比单图超分难得多，因为视频不仅空间维度需要复原，还需要对齐多帧之间的时间信息。BasicSR 实现了两个重量级视频算法：
这两个工作的意义在于：它们将视频超分从"针对特定算法的定制化实现"提升到了"通用框架"层面，同一个训练框架可以跑不同的视频复原任务。
除了超分辨率，BasicSR 还包含了图像去噪和去模糊功能。RIDNet（Real Image Denoising Network）可以在保留纹理细节的同时有效去除噪声；去模糊模块可以处理运动模糊和离焦模糊等多种退化类型。
BasicSR 提供了完整的 StyleGAN2 生成模型支持，可用于人脸复原和图像编辑。DFDNet（Deep Face Detector Network）专门针对低质量人脸图像进行复原，是 GFPGAN 等后续热门工具的基础模块。
ECBSR（Edge-oriented Convolution Block for Real-time Super Resolution on Mobile Devices）是 2021 年 ACM MM 会议的工作，专门为移动端实时超分设计。它通过精心设计的轻量级卷积模块，在手机等边缘设备上也能跑出不错的效果——这对需要在端侧部署 AI 的开发者来说非常有价值。
BasicSR 的代码架构非常清晰，核心模块分布在以下目录：
basicsr/
archs/ # 网络架构定义（EDSR、RCAN、SwinIR、RRDBNet 等）
models/ # 训练模型（SRGAN、ESRGAN、BasicVSR、SwinIR 等）
data/ # 数据集处理（LMDB、FileIO 等）
losses/ # 损失函数（Perceptual Loss、GAN Loss 等）
metrics/ # 评估指标（PSNR、SSIM、NIQE 等）
utils/ # 工具函数（Logger、Checkpoint、Learning Rate 等）
ops/ # 自定义算子（可能包含 CUDA 加速操作）
这种模块化设计的优势在于：换一个新算法，只需要继承基础模型类，重写网络结构文件，配置好训练选项文件（YAML），就可以复用整套训练、测试和推理流程。对于研究者来说，这大大降低了新想法的实验成本。
图2：BasicSR 工具箱整体概览
BasicSR 使用 YAML 配置文件管理训练参数，所有配置放在 options/ 目录下。训练一条命令即可启动：
python basicsr/train.py -opt options/your_config.yml
配置文件里包含数据集路径、模型架构、优化器设置、学习率调度等所有超参数。用户不需要改代码，只需要改配置文件。
推理则提供了 inference/ 目录下的专用脚本，每个脚本对应一个模型：
# 单图超分
python inference/inference_esrgan.py -i input_dir -o output_dir
# 视频超分
python inference/inference_basicvsr.py -i video.mp4 -o output_dir
# SwinIR
python inference/inference_swinir.py -i input.png -o output.png
BasicSR 还提供了 Model Zoo 页面，列出了所有预训练模型的下载链接。用户不需要从头训练，直接下载权重文件，放到指定目录，就能跑推理。
BasicSR 提供了 Google Colab 的集成，用户不需要本地有任何 GPU，只需要一个 Google 账号，点击链接就能在免费的 Colab TPU/GPU 上运行推理。这对非计算机背景的用户特别友好——不需要配置环境，不需要懂命令行，直接改参数就能用。
图3：BasicSR 训练过程可视化（W&B Dashboard）
BasicSR 的所有核心算法都依赖 NVIDIA GPU，CPU 推理速度极慢。对于没有 GPU 的用户，实际上无法正常使用。这与它"研究工具"的定位一致，但确实限制了它的受众范围。
项目没有提供 Dockerfile 或 docker-compose，意味着用户必须自己在主机上安装 PyTorch、CUDA、各种依赖。安装过程对新手来说并不友好，版本不兼容是常见坑。
视频超分需要逐帧处理，BasicVSR 的推理速度在消费级 GPU 上约为 5-10 FPS，难以满足实时应用需求。对于有实时需求的生产场景，需要进一步优化（如 TensorRT 加速），这部分 BasicSR 本身不提供支持。
BasicSR 不仅仅是一个算法集合，它代表了图像复原领域从"单算法独立实现"向"统一训练框架"演进的趋势。类似 mmdetection 在目标检测领域、Stable Diffusion 在图像生成领域所起的作用，BasicSR 为图像复原研究者提供了一个共享的基础设施。
截至目前，BasicSR 已在 GitHub 上获得超过 8200 颗星，fork 数超过 1400，issue 数量超过 350——这些数据表明它已经成为了该领域最受欢迎的基准工具之一。XPixel 团队持续活跃维护，版本已更新至 1.4.2。
随着 Transformer 架构在图像复原领域的深入应用（以 SwinIR 为代表），BasicSR 的架构也将继续演进。未来的方向可能包括：更高效的模型压缩与量化、与主流推理框架（ONNX、TensorRT）的深度集成，以及对更多视频编解码场景的原生支持。
对于 AI 研究者，BasicSR 是复现论文结果的必备工具；对于工程师，它提供了从训练到部署的完整 pipeline；对于 AI 爱好者，它是一个理解深度学习图像复原技术的优秀学习案例。