abc.ABC
spandrel.Architecture
- The abstract base class for all architectures.spandrel.architectures.ATD.ATDArch
- Undocumentedspandrel.architectures.Compact.CompactArch
- Undocumentedspandrel.architectures.CRAFT.CRAFTArch
- Undocumentedspandrel.architectures.DAT.DATArch
- Undocumentedspandrel.architectures.DCTLSA.DCTLSAArch
- Undocumentedspandrel.architectures.DITN.DITNArch
- Undocumentedspandrel.architectures.DnCNN.DnCNNArch
- Undocumentedspandrel.architectures.DRCT.DRCTArch
- Undocumentedspandrel.architectures.DRUNet.DRUNetArch
- Undocumentedspandrel.architectures.ESRGAN.ESRGANArch
- Undocumentedspandrel.architectures.FBCNN.FBCNNArch
- Undocumentedspandrel.architectures.FFTformer.FFTformerArch
- Undocumentedspandrel.architectures.GFPGAN.GFPGANArch
- Undocumentedspandrel.architectures.GRL.GRLArch
- Undocumentedspandrel.architectures.HAT.HATArch
- Undocumentedspandrel.architectures.IPT.IPTArch
- Undocumentedspandrel.architectures.KBNet.KBNetArch
- Undocumentedspandrel.architectures.LaMa.LaMaArch
- Undocumentedspandrel.architectures.MixDehazeNet.MixDehazeNetArch
- Undocumentedspandrel.architectures.MMRealSR.MMRealSRArch
- Undocumentedspandrel.architectures.NAFNet.NAFNetArch
- Undocumentedspandrel.architectures.OmniSR.OmniSRArch
- Undocumentedspandrel.architectures.PLKSR.PLKSRArch
- Undocumentedspandrel.architectures.RealCUGAN.RealCUGANArch
- Undocumentedspandrel.architectures.RestoreFormer.RestoreFormerArch
- Undocumentedspandrel.architectures.RGT.RGTArch
- Undocumentedspandrel.architectures.SAFMN.SAFMNArch
- Undocumentedspandrel.architectures.SAFMNBCIE.SAFMNBCIEArch
- Undocumentedspandrel.architectures.SCUNet.SCUNetArch
- Undocumentedspandrel.architectures.SPAN.SPANArch
- Undocumentedspandrel.architectures.SwiftSRGAN.SwiftSRGANArch
- Undocumentedspandrel.architectures.Swin2SR.Swin2SRArch
- Undocumentedspandrel.architectures.SwinIR.SwinIRArch
- Undocumentedspandrel.architectures.Uformer.UformerArch
- Undocumentedspandrel.ModelBase
- The base class of all model descriptors.spandrel.ImageModelDescriptor
- A model that takes an image as input and returns an image. See __call__
for more information.spandrel.MaskedImageModelDescriptor
- A model that takes an image and a mask for that image as input and returns an image. See __call__
for more information.enum.Enum
spandrel.ModelTiling
- Describes whether and how a model supports tiling.Exception
spandrel.UnsupportedDtypeError
- An error that will be thrown by .to
if the model does not support the given dtype.spandrel.UnsupportedModelError
- An error that will be thrown by ArchRegistry
and ModelLoader
if a model architecture is not supported.spandrel.ArchRegistry
- A registry of architectures.spandrel.ArchSupport
- An entry in an ArchRegistry
that describes how to detect and load a model architecture.spandrel.ModelLoader
- Class for automatically loading a pth file into any architecturespandrel.SizeRequirements
- A set of requirements for the size of an input image.spandrel.util.KeyCondition
- A condition that checks if a state dict has the given keys.torch.nn.Module
spandrel.architectures.__arch_helpers.block.ConcatBlock
- Undocumentedspandrel.architectures.__arch_helpers.block.ResidualDenseBlock_5C
- Residual Dense Block Style: 5 convs The core module of paper: (Residual Dense Network for Image Super-Resolution, CVPR 18) Modified options that can be used:spandrel.architectures.__arch_helpers.block.ResNetBlock
- ResNet Block, 3-3 style with extra residual scaling used in EDSR (Enhanced Deep Residual Networks for Single Image Super-Resolution, CVPRW 17)spandrel.architectures.__arch_helpers.block.RRDB
- Residual in Residual Dense Block (ESRGAN: Enhanced Super-Resolution Generative Adversarial Networks)spandrel.architectures.__arch_helpers.block.ShortcutBlock
- Undocumentedspandrel.util.timm.DropBlock2d
- DropBlock. See https://arxiv.org/pdf/1810.12890.pdfspandrel.util.timm.DropPath
- Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).typing.Generic
spandrel.Architecture
- The abstract base class for all architectures.spandrel.architectures.ATD.ATDArch
- Undocumentedspandrel.architectures.Compact.CompactArch
- Undocumentedspandrel.architectures.CRAFT.CRAFTArch
- Undocumentedspandrel.architectures.DAT.DATArch
- Undocumentedspandrel.architectures.DCTLSA.DCTLSAArch
- Undocumentedspandrel.architectures.DITN.DITNArch
- Undocumentedspandrel.architectures.DnCNN.DnCNNArch
- Undocumentedspandrel.architectures.DRCT.DRCTArch
- Undocumentedspandrel.architectures.DRUNet.DRUNetArch
- Undocumentedspandrel.architectures.ESRGAN.ESRGANArch
- Undocumentedspandrel.architectures.FBCNN.FBCNNArch
- Undocumentedspandrel.architectures.FFTformer.FFTformerArch
- Undocumentedspandrel.architectures.GFPGAN.GFPGANArch
- Undocumentedspandrel.architectures.GRL.GRLArch
- Undocumentedspandrel.architectures.HAT.HATArch
- Undocumentedspandrel.architectures.IPT.IPTArch
- Undocumentedspandrel.architectures.KBNet.KBNetArch
- Undocumentedspandrel.architectures.LaMa.LaMaArch
- Undocumentedspandrel.architectures.MixDehazeNet.MixDehazeNetArch
- Undocumentedspandrel.architectures.MMRealSR.MMRealSRArch
- Undocumentedspandrel.architectures.NAFNet.NAFNetArch
- Undocumentedspandrel.architectures.OmniSR.OmniSRArch
- Undocumentedspandrel.architectures.PLKSR.PLKSRArch
- Undocumentedspandrel.architectures.RealCUGAN.RealCUGANArch
- Undocumentedspandrel.architectures.RestoreFormer.RestoreFormerArch
- Undocumentedspandrel.architectures.RGT.RGTArch
- Undocumentedspandrel.architectures.SAFMN.SAFMNArch
- Undocumentedspandrel.architectures.SAFMNBCIE.SAFMNBCIEArch
- Undocumentedspandrel.architectures.SCUNet.SCUNetArch
- Undocumentedspandrel.architectures.SPAN.SPANArch
- Undocumentedspandrel.architectures.SwiftSRGAN.SwiftSRGANArch
- Undocumentedspandrel.architectures.Swin2SR.Swin2SRArch
- Undocumentedspandrel.architectures.SwinIR.SwinIRArch
- Undocumentedspandrel.architectures.Uformer.UformerArch
- Undocumentedspandrel.ImageModelDescriptor
- A model that takes an image as input and returns an image. See __call__
for more information.spandrel.MaskedImageModelDescriptor
- A model that takes an image and a mask for that image as input and returns an image. See __call__
for more information.spandrel.ModelBase
- The base class of all model descriptors.spandrel.ImageModelDescriptor
- A model that takes an image as input and returns an image. See __call__
for more information.spandrel.MaskedImageModelDescriptor
- A model that takes an image and a mask for that image as input and returns an image. See __call__
for more information.