MCPcopy Create free account

hub / github.com/allenai/satlas-super-resolution / types & classes

Types & classes26 in github.com/allenai/satlas-super-resolution

↓ 4 callersClassDoubleConv2d
Two-layer 2D convolutional block with a PReLU activation in between.
ssr/archs/arch_util.py:65
↓ 3 callersClassResidualDenseBlock
Residual Dense Block. Used in RRDB block in ESRGAN. Args: num_feat (int): Channel number of intermediate features. num_grow_
ssr/archs/rrdbnet_arch.py:14
↓ 2 callersClassFusionBlock
A block that fuses two revisits into one.
ssr/archs/arch_util.py:195
↓ 2 callersClassResidualBlock
Two-layer 2D convolutional block (DoubleConv2d) with a skip-connection to a sum.
ssr/archs/arch_util.py:136
↓ 2 callersClassSelfAttentionBlock
ssr/archs/osm_obj_discriminator_arch.py:8
↓ 1 callersClassCustomWeightedRandomSampler
WeightedRandomSampler except allows for more than 2^24 samples to be sampled. Source code: https://github.com/pytorch/pytorch/issues/2576#iss
ssr/data/s2-naip_dataset.py:18
↓ 1 callersClassHighResNet
High-resolution CNN. Inherits as many elements from SRCNN as possible for as fair a comparison: - DoubleConv2d: the in_channels are doubled b
ssr/archs/highresnet_arch.py:10
↓ 1 callersClassOneHot
One-hot encoder.
ssr/archs/arch_util.py:23
↓ 1 callersClassPixelShuffleBlock
PixelShuffle block with ConvTranspose2d for sub-pixel convolutions.
ssr/archs/arch_util.py:564
↓ 1 callersClassRecursiveFusion
Recursively fuses a set of encodings.
ssr/archs/arch_util.py:308
↓ 1 callersClassSRCNN
Super-resolution CNN. Uses no recursive function, revisits are treated as channels.
ssr/archs/srcnn_arch.py:13
↓ 1 callersClassSSR_RRDBNet
Networks consisting of Residual in Residual Dense Block, which is used in ESRGAN. ESRGAN: Enhanced Super-Resolution Generative Adversarial Ne
ssr/archs/rrdbnet_arch.py:72
ClassCLIPLoss
ssr/losses/basic_loss.py:20
ClassConvTransposeBlock
Upsampler block with ConvTranspose2d.
ssr/archs/arch_util.py:393
ClassDCNv2Pack
Modulated deformable conv for deformable alignment. Different from the official DCNv2Pack, which generates offsets and masks from the precedi
ssr/archs/arch_util.py:788
ClassDenseBlock
Two-layer 2D convolutional block (DoubleConv2d) with a skip-connection to a concatenation (instead of a sum used in ResidualBlock).
ssr/archs/arch_util.py:175
ClassL2Model
Wrapper model code to run the SRCNN and HighResNet architectures. Loss weights taken from the WorldStrat paper. Losses are hardcoded.
ssr/models/ssr_l2_model.py:15
ClassOSMObjDiscriminator
Arg: num_in_ch (int): Channel number of inputs. Default: 3. num_feat (int): Channel number of base intermediate features. Default
ssr/archs/osm_obj_discriminator_arch.py:35
ClassOSMObjESRGANModel
SSR ESRGAN Model: Training Satellite Imagery Super Resolution with Paired Training Data. The input to the generator is a time series of Sent
ssr/models/osm_objs_esrgan_model.py:23
ClassRRDB
Residual in Residual Dense Block. Used in RRDB-Net in ESRGAN. Args: num_feat (int): Channel number of intermediate features.
ssr/archs/rrdbnet_arch.py:47
ClassResidualBlockNoBN
Residual block without BN. Args: num_feat (int): Channel number of intermediate features. Default: 64. res_scale (flo
ssr/archs/arch_util.py:647
ClassS2NAIPDataset
Dataset object for the S2NAIP data. Builds a list of Sentinel-2 time series and NAIP image pairs. Args: opt (dict): Config for train
ssr/data/s2-naip_dataset.py:35
ClassSSIMLoss
ssr/losses/basic_loss.py:51
ClassSSRESRGANModel
SSR ESRGAN Model: Satellite imagery super-resolution model based on the Real-ESRGAN architecture. The input to the generator is a time serie
ssr/models/ssr_esrgan_model.py:19
ClassSSR_UNetDiscriminatorSN
Defines a U-Net discriminator with spectral normalization (SN) It is used in Real-ESRGAN: Training Real-World Blind Super-Resolution with Pure Sy
ssr/archs/discriminator_arch.py:12
ClassUpsample
Upsample module. Args: scale (int): Scale factor. Supported scales: 2^n and 3. num_feat (int): Channel number of intermediate fea
ssr/archs/arch_util.py:674