spandrel.architectures.SwiftSRGAN.SwiftSRGAN

class documentation

class SwiftSRGAN(nn.Module): (source)

Constructor: SwiftSRGAN(in_channels, num_channels, num_blocks, upscale_factor)

View In Hierarchy

Swift-SRGAN Generator

Parameters
in_channelsnumber of input image channels.
num_channelsnumber of hidden channels.
num_blocksnumber of residual blocks.
upscale_factorfactor to upscale the image [2x, 4x, 8x].
Returns
torch.Tensorsuper resolution image
Method __init__ Undocumented
Method forward Undocumented
Class Variable hyperparameters Undocumented
Instance Variable convblock Undocumented
Instance Variable final_conv Undocumented
Instance Variable initial Undocumented
Instance Variable residual Undocumented
Instance Variable upsampler Undocumented
def __init__(self, *, in_channels: int = 3, num_channels: int = 64, num_blocks: int = 16, upscale_factor: int = 4): (source)

Undocumented

def forward(self, x): (source)

Undocumented

hyperparameters: dict = (source)

Undocumented

convblock = (source)

Undocumented

final_conv = (source)

Undocumented

initial = (source)

Undocumented

residual = (source)

Undocumented

upsampler = (source)

Undocumented