Pytorch实现PixelShuffle

前言

​ 今天在思考如何解决model在上采样过程中可能会出现的棋盘格现象,阅读到一篇被CVPR2016收录的论文,里面提出了PixelShuffle,链接如下《Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network》

过程

​ PixelShuffle算法流程图如下,可以实现将hw的低分辨率图像,通过sub-pixel 操作变为rhrw的高分辨率图像,通过卷积层先得到了r^2个通道的图像,然后每个像素点的r^2个通道依次转换为对应的rr的图像块,最终得到rhrw的图像。

img

​ Pytorch中已经集成了PixelShuffle的module,torch.nn.PixelShuffle(upscale_factor),形参说明: - input (Variable) – 输入 - upscale_factor (int) – 增加空间分辨率的因子,例子如下。

img


   转载规则


《Pytorch实现PixelShuffle》 Severus 采用 知识共享署名 4.0 国际许可协议 进行许可。
 上一篇
爱如半夜汽笛 爱如半夜汽笛
《爱如半夜汽笛》—- 村上春树女孩问男孩:“你喜欢我喜欢到什么程度?” 少年想了想,用沉静的声音说:“半夜汽笛那个程度。” 少女默默地等待下文—里面肯定有什么故事。 “一次,半夜突然醒来。”他开始讲述,“确切时间不清楚,大约两三点吧,也
2019-08-12 Severus
下一篇 
Hello World Hello World
Welcome to Hexo! This is your very first post. Check documentation for more info. If you get any problems when using Hex
2019-08-10 Severus
  目录