HasnainRaz/Fast-SRGAN

A Fast Deep Learning Model to Upsample Low Resolution Videos to High Resolution at 30fps

解决的问题

解决低分辨率视频实时提升至高分辨率的挑战,提供比标准超分辨率方法更快的替代方案。

工作原理

该项目实现了 SR-GAN 架构,利用像素洗牌(pixel shuffle)进行高效上采样,从而提升处理速度。项目包含一个生成器模型,可基于 DIV2K 等数据集进行训练,也可直接使用预训练权重。

适用人群

寻找高性能超分辨率工具的开发者和研究人员,该工具能够以适合实时应用的速度处理视频帧(例如,实现 720p 输出时约 30fps)。

主要亮点

  • 实时性能:在 M1 Pro GPU 上,可实现约 27–30 FPS 的 720p 上采样。
  • 高效上采样:使用像素洗牌优化上采样过程。
  • 灵活训练:集成 Hydra,可通过命令行轻松配置超参数。
  • 预训练模型:包含在 DIV2K 数据集上预训练的生成器模型,可立即使用。

相关

  • 项目
  • 项目
  • 项目
  • 项目