Pytorch上采样函数 包括interpolate

本文介绍了PyTorch中的上采样操作,特别是使用interpolate函数进行2倍反卷积上采样、最近邻和双线性上采样。讨论了不同上采样方式的效果,如在ENet和MobileFaceNet中的应用,并指出在处理64x64输入时,Depth_Wise相比于MaxPool会慢10ms。通过interpolate函数可以方便地指定分辨率,解决奇数采样对齐问题。
摘要由CSDN通过智能技术生成

2倍反卷积上采样:

        self.trans_conv4 = nn.ConvTranspose2d(64, 64, kernel_size=3, stride=2, padding=1, output_padding=1, bias=False)
        self.stage3d = RSU4(128,16,64)

torch.nn的上下采样,可以任意分辨率:

最近邻上下采样:

import torch
from torch import nn

if __name__ == '__main__':

    import os
    data=torch.rand(2,16,16,16)
    upsample = nn.Upsample(size=(20,20), mode='nearest', align_corners=None)  # , mode='bilinear', align_corners=False)

    bbb=upsample(data)

    print(bbb.size())

双线性上下采样:

import torch
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

AI算法网奇

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值