1. 用法
torch.nn.Upsample(size=None, scale_factor=None, mode='nearest', align_corners=None, recompute_scale_factor=None)
- size: 指定输出的
尺寸
大小 - scale_factor: 指定输出的
尺寸
是输入尺寸
的倍数 - mode: 上采样的算法可选 ‘nearest’, ‘linear’, ‘bilinear’, ‘bicubic’,‘trilinear’. 默认: ‘nearest’
- align_corners和recomputer_scale_factor:几乎用不到
2. 避坑
这是尺寸的改变,不是通道改变,入门小白请注意! 是尺寸,尺寸,尺寸,不是通道,不是通道,不是通道
在Attention U-Net中有用到