【PyTorch】PixelShuffle

PyTorch中的PixelShuffle_gdymind的博客-CSDN博客_nn.pixelshuffle

定义

该类定义如下：

class torch.nn.PixleShuffle(upscale_factor)

这里的upscale_factor就是放大的倍数。

ps = nn.PixelShuffle(3)# 缩放到三倍，r == 3
input = torch.randn(1, 9, 4, 4)## r^2 C == 9，所以C == 1
output = ps(input)
print(output.size())
# 输出为：
# torch.Size([1, 1, 12, 12])

说白了就是使通道数降低，但是feature map变大，对于照片来说就是增加图像大小，降低通道数

【PyTorch】PixelShuffle相关推荐

【金融】【pytorch】使用深度学习预测期货收盘价涨跌——全连接神经网络模型构建与训练
[金融][pytorch]使用深度学习预测期货收盘价涨跌--全连接神经网络模型构建与训练模型构建与训练模型构建与训练 def get_accuracy(SR,GT,threshold=0.5):S ...
【金融】【pytorch】使用深度学习预测期货收盘价涨跌——LSTM模型构建与训练
[金融][pytorch]使用深度学习预测期货收盘价涨跌--LSTM模型构建与训练 LSTM 创建模型模型训练查看指标 LSTM 创建模型指标函数参考<如何用keras/tf/pytorc ...
【Pytorch】model.train() 和 model.eval() 原理与用法
文章目录一.两种模式二.功能 1. model.train() 2. model.eval() 为什么测试时要用 model.eval() ? 3. 总结与对比三.Dropout 简介参考链接 ...
【PyTorch】 tensor.squeeze压缩维度
[PyTorch] tensor.squeeze压缩维度 1.背景 2.squeeze方法 3.示例 1.背景 pytorch的tensor维度扩展.压缩是一个常见的操作, tensor维度扩展参见如 ...
【Pytorch】| Pytorch中softmax的dim的详细总结
[Pytorch]| Pytorch中softmax的dim的详细总结关于softmax的理解一维向量:dim=0和dim=-1结果相同,dim=1和dim=2会报错二维张量:dim=1和dim ...
【Pytorch】常见的人脸身份识别损失函数
[Pytorch]常见的人脸身份识别损失函数实验环境准备:人脸多角度多光照的图像数据集MUCT(276个受试者)+ MobileNetV3 说明:对于人脸身份数据集MUCT,是少样本数据集,应该使用 ...
【pytorch】过拟合的应对办法 —— 丢弃法（dropout）
文章目录一.什么是丢弃法,为什么丢弃法可以缓解过拟合? 二.丢弃法的手动实现三.丢弃法的pytorch实现参考关于过拟合.欠拟合的解释可以参考我的博文:[pytorch]过拟合和欠拟合详解,并 ...
【PyTorch】03数据处理
3 PyTorch数据处理 3.1 数据读取机制DataLoader与Dataset [PyTorch]2.1 DataLoader与Dataset epoch:对训练集的全部数据进行一次完整的训练, ...
【pytorch】——torch.hub
[pytorch]--torch.hub Facebook官方博客表示,PyTorch Hub是一个简易API和工作流程,为复现研究提供了基本构建模块,包含预训练模型库.并且,PyTorch Hub还 ...

【PyTorch】PixelShuffle

定义

【PyTorch】PixelShuffle相关推荐

最新文章

热门文章