【Pytorch】nn.PixelShuffle

作者：2023面试高手 | 2024-02-27 15:54:13

踩

nn.pixelshuffle

torch.nn.PixelShuffle(upscale_factor)
1

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
2
3
4
5
6
7
8
9
10
11

[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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/2023面试高手/article/detail/154238