spandrel.architectures.RetinexFormer.RetinexFormer

class documentation

class RetinexFormer(nn.Module): (source)

Constructor: RetinexFormer(in_channels, out_channels, n_feat, stage, num_blocks)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method forward x: [b,c,h,w] return out:[b,c,h,w]
Class Variable hyperparameters Undocumented
Instance Variable body Undocumented
Instance Variable stage Undocumented
def __init__(self, *, in_channels=3, out_channels=3, n_feat=40, stage=3, num_blocks=[1, 1, 1]): (source)

Undocumented

def forward(self, x): (source)

x: [b,c,h,w] return out:[b,c,h,w]

hyperparameters: dict = (source)

Undocumented

body = (source)

Undocumented

stage = (source)

Undocumented