2019.07.01|Pytorch实现PixelShuffle

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


PixelShuffle算法流程图

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

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


©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容