nn.Upsample() 参数详解及避坑

1. 用法

torch.nn.Upsample(size=None, scale_factor=None, mode='nearest', align_corners=None, recompute_scale_factor=None)
  1. size: 指定输出的尺寸大小
  2. scale_factor: 指定输出的尺寸是输入尺寸的倍数
  3. mode: 上采样的算法可选 ‘nearest’, ‘linear’, ‘bilinear’, ‘bicubic’,‘trilinear’. 默认: ‘nearest
  4. align_cornersrecomputer_scale_factor:几乎用不到

2. 避坑

这是尺寸的改变,不是通道改变,入门小白请注意! 是尺寸,尺寸,尺寸,不是通道,不是通道,不是通道
在Attention U-Net中有用到

猜你喜欢

转载自blog.csdn.net/qq_44864833/article/details/127916316