spandrel.architectures.__arch_helpers.dpir_basic_block

module documentation
(source)

Undocumented

Class ResBlock Undocumented
Function conv Undocumented
Function downsample_avgpool Undocumented
Function downsample_maxpool Undocumented
Function downsample_strideconv Undocumented
Function pixel_unshuffle Rearranges elements in a Tensor of shape (C, rH, rW) to a tensor of shape (*, r2C, H, W).
Function sequential Advanced nn.Sequential.
Function upsample_convtranspose Undocumented
Function upsample_pixelshuffle Undocumented
Function upsample_upconv Undocumented
def conv(in_channels=64, out_channels=64, kernel_size=3, stride=1, padding=1, bias=True, mode='CBR', negative_slope=0.2): (source)

Undocumented

def downsample_avgpool(in_channels=64, out_channels=64, kernel_size=3, stride=1, padding=1, bias=True, mode='2R', negative_slope=0.2): (source)

Undocumented

def downsample_maxpool(in_channels=64, out_channels=64, kernel_size=3, stride=1, padding=0, bias=True, mode='2R', negative_slope=0.2): (source)

Undocumented

def downsample_strideconv(in_channels=64, out_channels=64, kernel_size=2, stride=2, padding=0, bias=True, mode='2R', negative_slope=0.2): (source)

Undocumented

def pixel_unshuffle(input, upscale_factor): (source)

Rearranges elements in a Tensor of shape (C, rH, rW) to a tensor of shape (*, r2C, H, W).

Authors:
Zhaoyi Yan, https://github.com/Zhaoyi-Yan Kai Zhang, https://github.com/cszn/FFDNet
Date:
01/Jan/2019
def sequential(*args): (source)

Advanced nn.Sequential.

Parameters
*argsUndocumented
nn.Sequential
nn.Module
Returns
nn.Sequential
def upsample_convtranspose(in_channels=64, out_channels=3, kernel_size=2, stride=2, padding=0, bias=True, mode='2R', negative_slope=0.2): (source)

Undocumented

def upsample_pixelshuffle(in_channels=64, out_channels=3, kernel_size=3, stride=1, padding=1, bias=True, mode='2R', negative_slope=0.2): (source)

Undocumented

def upsample_upconv(in_channels=64, out_channels=3, kernel_size=3, stride=1, padding=1, bias=True, mode='2R', negative_slope=0.2): (source)

Undocumented