Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/PeterZhouSZ/StyleCariGAN
/ functions
Functions
131 in github.com/PeterZhouSZ/StyleCariGAN
⨍
Functions
131
◇
Types & classes
31
↓ 20 callers
Method
__init__
(self, kernel, pad, upsample_factor=1)
model.py:88
↓ 14 callers
Function
requires_grad
(model, flag=True, target_layer=None)
train.py:35
↓ 13 callers
Function
random_mat_apply
(p, transform, prev, eye)
non_leaking.py:143
↓ 11 callers
Method
backward
(ctx, grad_output)
op/upfirdn2d.py:127
↓ 10 callers
Function
mixing_noise
(batch, latent_dim, prob, device)
train.py:97
↓ 5 callers
Function
upfirdn2d
(input, kernel, up=1, down=1, pad=(0, 0))
op/upfirdn2d.py:145
↓ 4 callers
Function
d_logistic_loss
(real_pred, fake_pred)
train.py:50
↓ 4 callers
Function
d_r1_loss
(real_pred, real_img)
train.py:57
↓ 4 callers
Function
g_nonsaturating_loss
(fake_pred)
train.py:66
↓ 4 callers
Function
l2loss
(input1, input2)
invert.py:120
↓ 4 callers
Function
lognormal_sample
(size, mean=0, std=1)
non_leaking.py:120
↓ 4 callers
Function
uniform_sample
(size, low, high)
non_leaking.py:131
↓ 3 callers
Function
category_sample
(size, categories)
non_leaking.py:124
↓ 3 callers
Function
get_rank
()
distributed.py:9
↓ 3 callers
Function
get_world_size
()
distributed.py:34
↓ 3 callers
Function
make_kernel
(k)
model.py:34
↓ 3 callers
Function
rotate_mat
(theta)
non_leaking.py:35
↓ 3 callers
Function
scale_mat
(s_x, s_y)
non_leaking.py:47
↓ 2 callers
Function
accumulate
(model1, model2, decay=0.999)
train.py:42
↓ 2 callers
Function
fused_leaky_relu
(input, bias, negative_slope=0.2, scale=2 ** 0.5)
op/fused_act.py:86
↓ 2 callers
Function
invert
(g_ema, perceptual, real_img, device, args)
invert.py:131
↓ 2 callers
Function
make_noise
(batch, latent_dim, n_noise, device)
train.py:88
↓ 2 callers
Function
make_noise
(batch, latent_dim, n_noise, device)
invert.py:63
↓ 2 callers
Method
mean_latent
(self, n_latent)
model.py:524
↓ 2 callers
Function
normal_sample
(size, mean=0, std=1)
non_leaking.py:135
↓ 2 callers
Function
requires_grad
(model, flag=True)
invert.py:126
↓ 2 callers
Function
translate_mat
(t_x, t_y)
non_leaking.py:25
↓ 1 callers
Method
__init__
(self, device="cuda")
invert.py:50
↓ 1 callers
Method
__init__
( self, size, style_dim, n_mlp, channel_multiplier
exaggeration_model.py:15
↓ 1 callers
Function
affine_grid
(grid, mat)
non_leaking.py:254
↓ 1 callers
Function
apply_color
(img, mat)
non_leaking.py:374
↓ 1 callers
Function
bernoulli_sample
(size, p)
non_leaking.py:139
↓ 1 callers
Function
generate
( generator, truncation, truncation_latent, inversion_file, styles, device )
test.py:14
↓ 1 callers
Function
get_padding
(G, height, width)
non_leaking.py:259
↓ 1 callers
Method
load_cari_path
(self, path)
dataset_webcari.py:19
↓ 1 callers
Method
load_photo_path
(self, path)
dataset_webcari.py:29
↓ 1 callers
Function
luma_flip_mat
(axis, i)
non_leaking.py:99
↓ 1 callers
Function
make_grid
(shape, x0, x1, y0, y1, device)
non_leaking.py:244
↓ 1 callers
Function
mixing_noise
(batch, latent_dim, prob, device)
invert.py:72
↓ 1 callers
Function
noise_normalize_
(noises)
invert.py:99
↓ 1 callers
Function
noise_regularize
(noises)
invert.py:76
↓ 1 callers
Function
parse_args
()
invert.py:331
↓ 1 callers
Function
random_apply_affine
(img, p, G=None, antialiasing_kernel=SYM6)
non_leaking.py:316
↓ 1 callers
Function
random_apply_color
(img, p, C=None)
non_leaking.py:385
↓ 1 callers
Function
reduce_loss_dict
(loss_dict)
distributed.py:104
↓ 1 callers
Function
rotate3d_mat
(axis, theta)
non_leaking.py:67
↓ 1 callers
Function
sample_affine
(p, size, height, width)
non_leaking.py:151
↓ 1 callers
Function
sample_color
(p, size)
non_leaking.py:210
↓ 1 callers
Function
saturation_mat
(axis, i)
non_leaking.py:109
↓ 1 callers
Function
scale3d_mat
(s_x, s_y, s_z)
non_leaking.py:88
↓ 1 callers
Function
synchronize
()
distributed.py:19
↓ 1 callers
Function
train
(args, generator, discriminator_photo, discriminator_cari, discriminator_feat_p, discriminator_feat_c,
train.py:111
↓ 1 callers
Function
translate3d_mat
(t_x, t_y, t_z)
non_leaking.py:57
↓ 1 callers
Function
try_sample_affine_and_pad
(img, p, pad_k, G=None)
non_leaking.py:288
↓ 1 callers
Function
upfirdn2d_native
( input, kernel, up_x, up_y, down_x, down_y, pad_x0, pad_x1, pad_y0, pad_y1 )
op/upfirdn2d.py:159
Function
PYBIND11_MODULE
op/fused_bias_act.cpp:19
Function
PYBIND11_MODULE
op/upfirdn2d.cpp:21
Method
__call__
(self, t)
invert.py:56
Method
__getitem__
(self, index)
dataset_webcari.py:42
Method
__init__
(self)
invert.py:18
Method
__init__
(self, path, transform, resolution=8)
dataset_webcari.py:9
Method
__init__
( self, size, style_dim, n_mlp, channel_multiplier=2, blur_ker
exaggeration_model.py:568
Method
__init__
(self)
model.py:16
Method
__init__
(self)
model.py:27
Method
__init__
(self, kernel, factor=2)
model.py:46
Method
__init__
(self, kernel, factor=2)
model.py:67
Method
__init__
( self, in_channel, out_channel, kernel_size, stride=1, padding=0, bias=True )
model.py:107
Method
__init__
( self, in_dim, out_dim, bias=True, bias_init=0, lr_mul=1, activation=None )
model.py:145
Method
__init__
(self, negative_slope=0.2)
model.py:182
Method
__init__
( self, in_channel, out_channel, kernel_size, style_dim, demod
model.py:194
Method
__init__
(self)
model.py:293
Method
__init__
(self, channel, size=4)
model.py:307
Method
__init__
( self, in_channel, out_channel, kernel_size, style_dim, upsam
model.py:320
Method
__init__
(self, in_channel, style_dim, upsample=True, blur_kernel=[1, 3, 3, 1])
model.py:357
Method
__init__
(self, in_channel, out_channel, kernel_size=3, padding=1)
model.py:378
Method
__init__
(self, in_channel, out_channel, kernel_size=3, padding=1)
model.py:396
Method
__init__
(self, n=3)
model.py:416
Method
__init__
( self, size, style_dim, n_mlp, channel_multiplier=2, blur_ker
model.py:428
Method
__init__
( self, in_channel, out_channel, kernel_size, downsample=False,
model.py:628
Method
__init__
(self, in_channel, out_channel, blur_kernel=[1, 3, 3, 1])
model.py:677
Method
__init__
(self, size, channel_multiplier=2, blur_kernel=[1, 3, 3, 1])
model.py:698
Method
__init__
(self, size, channel_multiplier=2, blur_kernel=[1, 3, 3, 1])
model.py:758
Method
__init__
(self, channel, negative_slope=0.2, scale=2 ** 0.5)
op/fused_act.py:75
Method
__len__
(self)
dataset_webcari.py:39
Method
__repr__
(self)
model.py:137
Method
__repr__
(self)
model.py:175
Method
__repr__
(self)
model.py:242
Function
all_gather
(data)
distributed.py:69
Function
augment
(img, p, transform_matrix=(None, None))
non_leaking.py:394
Method
backward
(ctx, gradgrad_input, gradgrad_bias)
op/fused_act.py:43
Method
backward
(ctx, grad_output)
op/fused_act.py:64
Method
backward
(ctx, gradgrad_input)
op/upfirdn2d.py:63
Method
forward
(self, x)
invert.py:34
Method
forward
( self, styles, styles_fine=None, return_latents=False, inject_index=N
exaggeration_model.py:48
Method
forward
( self, styles, return_latents=False, inject_index=None, truncation=1,
exaggeration_model.py:675
Method
forward
(self, x)
model.py:22
Method
forward
(self, input)
model.py:30
Method
forward
(self, input)
model.py:60
Method
forward
(self, input)
model.py:81
Method
forward
(self, input)
model.py:100
next →
1–100 of 131, ranked by callers