spandrel.architectures.MMRealSR.MMRealSR

class documentation

class MMRealSR(nn.Module): (source)

Constructor: MMRealSR(num_in_ch, num_out_ch, scale, num_feat, ...)

Networks consisting of Residual in Residual Dense Block, which is used in ESRGAN. v2.1

ESRGAN: Enhanced Super-Resolution Generative Adversarial Networks.

We extend ESRGAN for scale x2 and scale x1. Note: This is one option for scale 1, scale 2 in RRDBNet. We first employ the pixel-unshuffle (an inverse operation of pixelshuffle to reduce the spatial size and enlarge the channel size before feeding inputs into the main ESRGAN architecture.

Parameters
num_in_ch	Channel number of inputs.
num_out_ch	Channel number of outputs.
num_feat	Channel number of intermediate features. Default: 64
num_block	Block number in the trunk network. Defaults: 23
num_grow_ch	Channels for each growth. Default: 32.

Method	`__init__`	Undocumented
Method	`forward`	Undocumented
Class Variable	`hyperparameters`	Undocumented
Instance Variable	`am_list`	Undocumented
Instance Variable	`body`	Undocumented
Instance Variable	`conv_body`	Undocumented
Instance Variable	`conv_first`	Undocumented
Instance Variable	`conv_hr`	Undocumented
Instance Variable	`conv_last`	Undocumented
Instance Variable	`conv_up1`	Undocumented
Instance Variable	`conv_up2`	Undocumented
Instance Variable	`dd_embed`	Undocumented
Instance Variable	`de_net`	Undocumented
Instance Variable	`lrelu`	Undocumented
Instance Variable	`num_block`	Undocumented
Instance Variable	`num_degradation`	Undocumented
Instance Variable	`scale`	Undocumented

def __init__(self, *, num_in_ch, num_out_ch, scale=4, num_feat=64, num_block=23, num_grow_ch=32, de_net_type='DEResNet', num_degradation=2, degradation_degree_actv='sigmoid', num_feats=[64, 128, 256, 512], num_blocks=[2, 2, 2, 2], downscales=[2, 2, 2, 1]): (source) ¶

Undocumented

def forward(self, x, custom_degrees=(None, None)): (source) ¶

Undocumented

hyperparameters: dict = (source) ¶

Undocumented

am_list = (source) ¶

Undocumented

body = (source) ¶

Undocumented

conv_body = (source) ¶

Undocumented

conv_first = (source) ¶