mmagic.models.archs.upsample
¶
Module Contents¶
Classes¶
Pixel Shuffle upsample layer. |
- class mmagic.models.archs.upsample.PixelShufflePack(in_channels: int, out_channels: int, scale_factor: int, upsample_kernel: int)[源代码]¶
Bases:
torch.nn.Module
Pixel Shuffle upsample layer.
- 参数
in_channels (int) – Number of input channels.
out_channels (int) – Number of output channels.
scale_factor (int) – Upsample ratio.
upsample_kernel (int) – Kernel size of Conv layer to expand channels.
- 返回
Upsampled feature map.