【Pytorch】nn.PixelShuffle

二进制人工智能

已于 2023-06-12 16:23:04 修改

阅读量2.6k

点赞数 5

分类专栏： pytorch 文章标签： pytorch 深度学习 python

于 2022-08-10 17:33:29 首次发布

本文链接：https://blog.csdn.net/weixin_44378835/article/details/126267016

版权

pytorch 专栏收录该内容

20 篇文章 18 订阅

订阅专栏

torch.nn.PixelShuffle(upscale_factor)

PixelShuffle是一种上采样方法，它将形状 $C\times r^2, H, W)$ 的张量重新排列转换为形状为 $H\times r, W\times r)$ 的张量：

图片来源：[1]

其中 $r$ 是upscale_factor因子。

输入输出尺寸：

例子： $2)\rightarrow (1,2,4,4)$

import torch
import torch.nn as nn

ps = nn.PixelShuffle(2)
input = torch.arange(0, 8 * 2 * 2).view(1, 8, 2, 2)
output = ps(input)

print('input:\n',input)
print('output:\n',output)

[1] Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network https://openaccess.thecvf.com/content_cvpr_2016/papers/Shi_Real-Time_Single_Image_CVPR_2016_paper.pdf

实现：
https://blog.csdn.net/qq249356520/article/details/95318645