PaperNote-Fast and Accurate Single Image Super-Resolution via Information Distillation Network

PaperNote-Fast and Accurate Single Image Super-Resolution via Information Distillation Network

论文地址:https://arxiv.org/pdf/1803.09454.pdf

代码地址:https://github.com/Zheng222/IDN-Caffe

论文主要提出了IDN网络(Information Distillation Network

网络主要结构:

PaperNote-Fast and Accurate Single Image Super-Resolution via Information Distillation Network

IDN主要分为三个部分:

特征抽取块(feature extraction block) 它主要由两个卷积层叠加而成。

信息蒸馏块(information distillation block) 由两个小function构成 Enhancement unit,由三个3*3的卷积构成,每一个后面跟着一个LRelu**函数。Compression unit,由一个1*1的卷积层构成,主要将Enhancement unit蒸馏后的信息进行融合。

PaperNote-Fast and Accurate Single Image Super-Resolution via Information Distillation Network

图像重构块(RBlock) 由一个上采样卷积构成。

训练细节:采用MAE作为loss先进行训练然后采用MSE进行finetune。将训练集291图像resize为78*78,然后采用缩小后经上采样放大方式进行训练。网络的主要层数为31层,有4个DBlocks。每个enhancement unit通道依次为64,16,4,为了减少参数采用分组卷积,LRelu初始化alpha为0.05,batch_size 为64,学习率为 1e-4.

结果:

PaperNote-Fast and Accurate Single Image Super-Resolution via Information Distillation Network