4 个月前

使用高效亚像素卷积神经网络实现实时单图像和视频超分辨率

使用高效亚像素卷积神经网络实现实时单图像和视频超分辨率

摘要

最近,基于深度神经网络的几种模型在单图像超分辨率的重建精度和计算性能方面取得了显著成功。在这些方法中,低分辨率(LR)输入图像通常通过单一滤波器(通常是双三次插值)放大到高分辨率(HR)空间,然后再进行重建。这意味着超分辨率(SR)操作是在HR空间中执行的。我们证明了这种方法次优且增加了计算复杂度。本文提出了一种能够在单个K2 GPU上实现实时1080p视频超分辨率的卷积神经网络(CNN)。为了实现这一目标,我们设计了一种新颖的CNN架构,在该架构中特征图是在LR空间中提取的。此外,我们引入了一种高效的子像素卷积层,该层学习一系列放大滤波器,将最终的LR特征图放大为HR输出。通过这种方式,我们将SR流程中的手工设计的双三次滤波器替换为针对每个特征图专门训练的更复杂的放大滤波器,同时降低了整体SR操作的计算复杂度。我们使用公开数据集中的图像和视频对所提出的方案进行了评估,并显示其性能显著优于先前基于CNN的方法(图像提升+0.15 dB,视频提升+0.39 dB),并且速度提高了近一个数量级。

代码仓库

Nhat-Thanh/ESPCN-Pytorch
pytorch
GitHub 中提及
kingcheng2000/GAN
GitHub 中提及
npielawski/pytorch_tiramisu
pytorch
GitHub 中提及
poikilos/pyrotocanvas
tf
GitHub 中提及
niazwazir/REAL_TIME_VIDEO_IMSR
pytorch
GitHub 中提及
med-seg/kidney
tf
GitHub 中提及
HighVoltageRocknRoll/sr
tf
GitHub 中提及
atriumlts/subpixel
tf
GitHub 中提及
anujdutt9/ESPCN
pytorch
GitHub 中提及
gs18113/ESPCN-TensorFlow2
tf
GitHub 中提及
alexjc/neural-enhance
GitHub 中提及
TahmasbiM/Example
tf
GitHub 中提及
KeremTurgutlu/papers
GitHub 中提及
michael13162/DoodleGAN
GitHub 中提及
jaingaurav3/GAN-Hacks
GitHub 中提及
leftthomas/espcn
pytorch
GitHub 中提及
Nhat-Thanh/ESPCN-TF
tf
GitHub 中提及
Araxeus/PNG-Upscale
tf
GitHub 中提及
quin-med-harvard-edu/kidney
tf
GitHub 中提及
twhui/SRGAN-PyTorch
pytorch
GitHub 中提及
ssulun/pytorch-pixelshuffle1d
pytorch
GitHub 中提及
med-seg/kidney-mc
tf
GitHub 中提及
vuanhtu1993/Keras-SRGANs
tf
GitHub 中提及
TomokiKomiya/SRGAN-keras
tf
GitHub 中提及
tetrachrome/subpixel
tf
GitHub 中提及
Zhangyanbo/iResNetLab
pytorch
GitHub 中提及
545641826/espcn
pytorch
GitHub 中提及
deepak112/Keras-SRGAN
tf
GitHub 中提及
fengwang/subpixel_conv2d
tf
GitHub 中提及
Lornatang/ESPCN-PyTorch
pytorch
GitHub 中提及

基准测试

基准方法指标
image-super-resolution-on-bsd100-4x-upscalingESPCN
MOS: 2.01
PSNR: 27.02
SSIM: 0.7442
image-super-resolution-on-set14-4x-upscalingESPCN
MOS: 2.52
PSNR: 27.66
SSIM: 0.8004
video-super-resolution-on-msu-video-upscalersESPCN
PSNR: 26.25
SSIM: 0.926
VMAF: 47.19
video-super-resolution-on-msu-vsr-benchmarkESPCN
1 - LPIPS: 0.765
ERQAv1.0: 0.521
FPS: 3.333
PSNR: 26.714
QRCRv1.0: 0
SSIM: 0.811
Subjective score: 2.099
video-super-resolution-on-ultra-video-groupESPCN
Average PSNR: 37.91
video-super-resolution-on-ultra-video-groupbicubic
Average PSNR: 36.20
video-super-resolution-on-vid4-4x-upscalingESPCN
MOVIE: 6.54
PSNR: 25.06
SSIM: 0.7394
video-super-resolution-on-xiph-hd-4xESPCN
Average PSNR: 31.67
video-super-resolution-on-xiph-hd-4xbicubic
Average PSNR: 30.30

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
使用高效亚像素卷积神经网络实现实时单图像和视频超分辨率 | 论文 | HyperAI超神经