Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/JCBrouwer/maua-stylegan2
/ types & classes
Types & classes
120 in github.com/JCBrouwer/maua-stylegan2
⨍
Functions
576
◇
Types & classes
120
↓ 62 callers
Class
conv2DBatchNormRelu
models/autoencoder.py:143
↓ 15 callers
Class
Generator
models/stylegan2.py:368
↓ 13 callers
Class
RandomApply
contrastive_learner.py:76
↓ 12 callers
Class
FusedLeakyReLU
op/fused_act.py:74
↓ 10 callers
Class
MyLinear
Linear layer with equalized learning rate and custom learning rate multiplier.
models/stylegan1.py:12
↓ 7 callers
Class
MultiResolutionDataset
dataset.py:10
↓ 7 callers
Class
NetLinLayer
A single linear layer which does a 1x1 conv
validation/lpips/networks_basic.py:110
↓ 6 callers
Class
InceptionV3
Pretrained InceptionV3 network returning feature maps
validation/inception.py:18
↓ 5 callers
Class
ConvLayer
models/stylegan2.py:579
↓ 5 callers
Class
EqualLinear
models/stylegan2.py:123
↓ 4 callers
Class
DecoderModule
models/autoencoder.py:595
↓ 4 callers
Class
Discriminator
models/stylegan2.py:646
↓ 4 callers
Class
EncoderModule
models/autoencoder.py:513
↓ 4 callers
Class
FIDInceptionC
InceptionC block patched for FID computation
validation/inception.py:213
↓ 4 callers
Class
G_synthesis
models/stylegan1.py:418
↓ 4 callers
Class
LayerEpilogue
Things to do at the end of each layer.
models/stylegan1.py:238
↓ 4 callers
Class
MyConv2d
Conv layer with equalized learning rate and custom learning rate multiplier.
models/stylegan1.py:41
↓ 3 callers
Class
Blur
models/stylegan2.py:76
↓ 3 callers
Class
ContrastiveLearner
contrastive_learner.py:156
↓ 3 callers
Class
Conv2DMod
lucidrains.py:305
↓ 3 callers
Class
FIDInceptionA
InceptionA block patched for FID computation
validation/inception.py:188
↓ 3 callers
Class
Scrolling_Area
select_latents.py:137
↓ 3 callers
Class
StyledConv
models/stylegan2.py:310
↓ 3 callers
Class
segnetDown3
models/autoencoder.py:186
↓ 3 callers
Class
segnetUp3
models/autoencoder.py:217
↓ 2 callers
Class
AddNoise
Adds static noise to output Args: noise (th.tensor): Noise to be added
audioreactive/bend.py:28
↓ 2 callers
Class
ConvSegNet
Adapted from https://github.com/foamliu/Autoencoder See LICENSE_AUTOENCODER
models/autoencoder.py:337
↓ 2 callers
Class
EMA
lucidrains.py:63
↓ 2 callers
Class
G_style
models/stylegan1.py:509
↓ 2 callers
Class
Generator
lucidrains.py:401
↓ 2 callers
Class
HoverButton
select_latents.py:29
↓ 2 callers
Class
InvisibleScrollbar
select_latents.py:43
↓ 2 callers
Class
LookaheadMinimax
r""" A PyTorch implementation of the lookahead wrapper for GANs. This optimizer performs the lookahead step on both the discriminator and gen
lookahead_minimax.py:7
↓ 2 callers
Class
ManipulationLayer
models/stylegan2.py:297
↓ 2 callers
Class
ModulatedConv2d
models/stylegan2.py:164
↓ 2 callers
Class
PixelNormLayer
models/stylegan1.py:139
↓ 2 callers
Class
Residual
lucidrains.py:79
↓ 2 callers
Class
Rezero
lucidrains.py:88
↓ 2 callers
Class
StyleVectorizer
lucidrains.py:266
↓ 2 callers
Class
ToRGB
models/stylegan2.py:346
↓ 2 callers
Class
segnetDown2
models/autoencoder.py:171
↓ 2 callers
Class
segnetUp2
models/autoencoder.py:203
↓ 1 callers
Class
BlurLayer
models/stylegan1.py:148
↓ 1 callers
Class
ConstantInput
models/stylegan2.py:269
↓ 1 callers
Class
Dataset
lucidrains.py:232
↓ 1 callers
Class
Decoder
models/autoencoder.py:608
↓ 1 callers
Class
DecoderInceptionModuleSingle
models/autoencoder.py:572
↓ 1 callers
Class
Discriminator
lucidrains.py:449
↓ 1 callers
Class
DiscriminatorBlock
lucidrains.py:378
↓ 1 callers
Class
Dist2LogitLayer
takes 2 distances, puts through fc layers, spits out value between [0,1] (if use_sigmoid is True)
validation/lpips/networks_basic.py:123
↓ 1 callers
Class
EMA
contrastive_learner.py:95
↓ 1 callers
Class
Encoder
models/autoencoder.py:526
↓ 1 callers
Class
EncoderInceptionModuleSignle
models/autoencoder.py:490
↓ 1 callers
Class
EqualConv2d
models/stylegan2.py:95
↓ 1 callers
Class
FIDInceptionE_1
First InceptionE block patched for FID computation
validation/inception.py:241
↓ 1 callers
Class
FIDInceptionE_2
Second InceptionE block patched for FID computation
validation/inception.py:274
↓ 1 callers
Class
Flatten
lucidrains.py:74
↓ 1 callers
Class
Flatten
models/autoencoder.py:26
↓ 1 callers
Class
GSynthesisBlock
models/stylegan1.py:364
↓ 1 callers
Class
G_mapping
models/stylegan1.py:191
↓ 1 callers
Class
GeneratorBlock
lucidrains.py:342
↓ 1 callers
Class
InceptionVAE
Adapted from https://github.com/koshian2/inception-vae
models/autoencoder.py:639
↓ 1 callers
Class
InputBlock
models/stylegan1.py:318
↓ 1 callers
Class
LatentInput
models/stylegan2.py:281
↓ 1 callers
Class
LogCoshVAE
Adapted from https://github.com/AntixK/PyTorch-VAE See LICENSE_AUTOENCODER
models/autoencoder.py:41
↓ 1 callers
Class
Mousewheel_Support
select_latents.py:49
↓ 1 callers
Class
NoiseInjection
models/stylegan2.py:257
↓ 1 callers
Class
NoiseLayer
adds noise. noise is per pixel (constant over channels) with per-channel weight
models/stylegan1.py:108
↓ 1 callers
Class
OutputHiddenLayer
contrastive_learner.py:115
↓ 1 callers
Class
PermuteToFrom
lucidrains.py:98
↓ 1 callers
Class
PixelNorm
models/stylegan2.py:15
↓ 1 callers
Class
RGBBlock
lucidrains.py:280
↓ 1 callers
Class
ResBlock
models/stylegan2.py:623
↓ 1 callers
Class
ScaledLeakyReLU
models/stylegan2.py:152
↓ 1 callers
Class
ScalingLayer
validation/lpips/networks_basic.py:100
↓ 1 callers
Class
SpectralNorm
validation/spectral_norm.py:4
↓ 1 callers
Class
StyleGAN2
lucidrains.py:512
↓ 1 callers
Class
StyleGAN2
lightning.py:33
↓ 1 callers
Class
StyleMod
models/stylegan1.py:126
↓ 1 callers
Class
Trainer
lucidrains.py:625
↓ 1 callers
Class
UnFlatten
models/autoencoder.py:31
↓ 1 callers
Class
Upsample
models/stylegan2.py:34
↓ 1 callers
Class
Upscale2d
models/stylegan1.py:180
↓ 1 callers
Class
VGG19
Adapted from https://github.com/NVIDIA/pix2pixHD See LICENSE-VGG
accelerate/accelerate_inception.py:24
↓ 1 callers
Class
VGG19
Adapted from https://github.com/NVIDIA/pix2pixHD See LICENSE-VGG
accelerate/accelerate_logcosh.py:29
↓ 1 callers
Class
VGG19
Adapted from https://github.com/NVIDIA/pix2pixHD See LICENSE-VGG
accelerate/accelerate_segnet.py:23
↓ 1 callers
Class
VGGLoss
Adapted from https://github.com/NVIDIA/pix2pixHD See LICENSE-VGG
accelerate/accelerate_inception.py:62
↓ 1 callers
Class
VGGLoss
Adapted from https://github.com/NVIDIA/pix2pixHD See LICENSE-VGG
accelerate/accelerate_logcosh.py:67
↓ 1 callers
Class
VGGLoss
Adapted from https://github.com/NVIDIA/pix2pixHD See LICENSE-VGG
accelerate/accelerate_segnet.py:61
↓ 1 callers
Class
addNoise
generate_video.py:462
↓ 1 callers
Class
expand_greyscale
lucidrains.py:218
Class
BCERankingLoss
validation/lpips/networks_basic.py:154
Class
BaseModel
validation/lpips/base_model.py:5
Class
DSSIM
validation/lpips/networks_basic.py:197
Class
DistModel
validation/lpips/dist_model.py:15
Class
Downsample
models/stylegan2.py:55
Class
FakeNet
validation/lpips/networks_basic.py:168
Class
FusedLeakyReLUFunction
op/fused_act.py:52
Class
FusedLeakyReLUFunctionBackward
op/fused_act.py:20
Class
Inception3Feature
validation/calc_inception.py:18
next →
1–100 of 120, ranked by callers