spandrel.architectures.MixDehazeNet.MixDehazeNet

class documentation

class MixDehazeNet(nn.Module): (source)

Constructor: MixDehazeNet(in_chans, out_chans, embed_dims, depths)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method check_image_size Undocumented
Method forward Undocumented
Method forward_features Undocumented
Class Variable hyperparameters Undocumented
Instance Variable fusion1 Undocumented
Instance Variable fusion2 Undocumented
Instance Variable layer1 Undocumented
Instance Variable layer2 Undocumented
Instance Variable layer3 Undocumented
Instance Variable layer4 Undocumented
Instance Variable layer5 Undocumented
Instance Variable patch_embed Undocumented
Instance Variable patch_merge1 Undocumented
Instance Variable patch_merge2 Undocumented
Instance Variable patch_size Undocumented
Instance Variable patch_split1 Undocumented
Instance Variable patch_split2 Undocumented
Instance Variable patch_unembed Undocumented
Instance Variable skip1 Undocumented
Instance Variable skip2 Undocumented
def __init__(self, *, in_chans=3, out_chans=4, embed_dims=[24, 48, 96, 48, 24], depths=[1, 1, 2, 1, 1]): (source)

Undocumented

def check_image_size(self, x): (source)

Undocumented

def forward(self, x): (source)

Undocumented

def forward_features(self, x): (source)

Undocumented

hyperparameters: dict = (source)

Undocumented

fusion1 = (source)

Undocumented

fusion2 = (source)

Undocumented

layer1 = (source)

Undocumented

layer2 = (source)

Undocumented

layer3 = (source)

Undocumented

layer4 = (source)

Undocumented

layer5 = (source)

Undocumented

patch_embed = (source)

Undocumented

patch_merge1 = (source)

Undocumented

patch_merge2 = (source)

Undocumented

patch_size: int = (source)

Undocumented

patch_split1 = (source)

Undocumented

patch_split2 = (source)

Undocumented

patch_unembed = (source)

Undocumented

skip1 = (source)

Undocumented

skip2 = (source)

Undocumented