Functions — github.com/clovaai/stargan-v2

hub / github.com/clovaai/stargan-v2 / functions

Functions126 in github.com/clovaai/stargan-v2

⨍ Functions 126◇ Types & classes 23

↓ 9 callersMethodget_heatmap

outputs 0-1 normalized heatmap

core/wing.py:249

↓ 7 callersMethod__init__

(self, style_dim, num_features)

core/model.py:68

↓ 6 callersMethodload

(self, step)

core/checkpoint.py:37

↓ 5 callersFunctiontqdm

(x)

metrics/fid.py:24

↓ 4 callersMethod__init__

(self, num_modules=1, end_relu=False, num_landmarks=98, fname_pretrained=None)

core/wing.py:191

↓ 4 callersMethod_reset_grad

(self)

core/solver.py:75

↓ 4 callersFunctioncalculate_metrics

(nets, args, step, mode)

metrics/eval.py:26

↓ 4 callersFunctionresize

Resize heatmaps.

core/wing.py:454

↓ 4 callersFunctionsave_image

(x, ncol, filename)

core/utils.py:57

↓ 4 callersFunctionshift

Shift N pixels up or down.

core/wing.py:459

↓ 4 callersFunctionsubdirs

(dname)

main.py:27

↓ 3 callersMethod_load_checkpoint

(self, step)

core/solver.py:71

↓ 3 callersFunctionadv_loss

(logits, target)

core/solver.py:275

↓ 3 callersFunctionget_eval_loader

(root, img_size=256, batch_size=32, imagenet_normalize=True, shuffle=True,

core/data_loader.py:123

↓ 3 callersFunctionget_test_loader

(root, img_size=256, batch_size=32, shuffle=True, num_workers=4)

core/data_loader.py:152

↓ 3 callersFunctionmoving_average

(model, model_test, beta=0.999)

core/solver.py:270

↓ 2 callersMethod__init__

(self)

metrics/lpips.py:50

↓ 2 callersMethod_fetch_inputs

(self)

core/data_loader.py:178

↓ 2 callersFunctioncalculate_fid_given_paths

(paths, img_size=256, batch_size=50)

metrics/fid.py:63

↓ 2 callersFunctioncompute_d_loss

(nets, args, x_real, y_org, y_trg, z_trg=None, x_ref=None, masks=None)

core/solver.py:201

↓ 2 callersFunctioncompute_g_loss

(nets, args, x_real, y_org, y_trg, z_trgs=None, x_refs=None, masks=None)

core/solver.py:226

↓ 2 callersFunctionget_alphas

(start=-5, end=5, step=0.5, len_tail=10)

core/utils.py:155

↓ 2 callersFunctionget_train_loader

(root, which='source', img_size=256, batch_size=8, prob=0.5, num_workers=4)

core/data_loader.py:88

↓ 2 callersFunctioninterpolate

returns T x C x H x W

core/utils.py:159

↓ 2 callersFunctionlandmarks2xaxis

(landmarks)

core/wing.py:351

↓ 2 callersFunctionlistdir

(dname)

core/data_loader.py:27

↓ 2 callersFunctionnormalize

(x, eps=1e-10)

metrics/lpips.py:16

↓ 2 callersFunctionpoints2T

(point, direction)

core/wing.py:324

↓ 2 callersMethodsample

(self, loaders)

core/solver.py:174

↓ 2 callersFunctionsave_video

(fname, images, output_fps=30, vcodec='libx264', filters='')

core/utils.py:262

↓ 2 callersFunctiontensor2ndarray255

(images)

core/utils.py:277

↓ 1 callersMethod_build_weights

(self, dim_in, dim_out)

core/model.py:33

↓ 1 callersMethod_build_weights

(self, dim_in, dim_out, style_dim=64)

core/model.py:90

↓ 1 callersMethod_fetch_refs

(self)

core/data_loader.py:186

↓ 1 callersMethod_forward

(self, level, inp)

core/wing.py:69

↓ 1 callersMethod_generate_network

(self, level)

core/wing.py:60

↓ 1 callersMethod_load_lpips_weights

(self)

metrics/lpips.py:61

↓ 1 callersFunction_make_balanced_sampler

(labels)

core/data_loader.py:81

↓ 1 callersMethod_make_dataset

(self, root)

core/data_loader.py:56

↓ 1 callersMethod_residual

(self, x)

core/model.py:49

↓ 1 callersMethod_residual

(self, x, s)

core/model.py:105

↓ 1 callersMethod_save_checkpoint

(self, step)

core/solver.py:67

↓ 1 callersMethod_shortcut

(self, x)

core/model.py:42

↓ 1 callersMethod_shortcut

(self, x)

core/model.py:98

↓ 1 callersMethodalign

imgs = torch.CUDATensor of BCHW

core/wing.py:299

↓ 1 callersFunctionalign_faces

(args, input_dir, output_dir)

core/wing.py:407

↓ 1 callersFunctionbuild_model

(args)

core/model.py:283

↓ 1 callersFunctioncalculate_fid_for_all_tasks

(args, domains, step, mode)

metrics/eval.py:122

↓ 1 callersFunctioncalculate_lpips_given_images

(group_of_images)

metrics/lpips.py:86

↓ 1 callersFunctiondenormalize

(x)

core/utils.py:52

↓ 1 callersMethodevaluate

(self)

core/solver.py:192

↓ 1 callersFunctionfrechet_distance

(mu, cov, mu2, cov2)

metrics/fid.py:56

↓ 1 callersMethodget_landmark

outputs landmarks of x.shape

core/wing.py:263

↓ 1 callersFunctionget_preds_fromhm

(hm)

core/wing.py:27

↓ 1 callersFunctionlandmarks2S

(x, y)

core/wing.py:371

↓ 1 callersFunctionlandmarks2eyes

(landmarks)

core/wing.py:332

↓ 1 callersMethodlandmarks2mat

(self, landmarks)

core/wing.py:314

↓ 1 callersFunctionlandmarks2mouthends

(landmarks)

core/wing.py:340

↓ 1 callersMethodload_pretrained_weights

(self, fname)

core/wing.py:215

↓ 1 callersFunctionmain

(args)

main.py:32

↓ 1 callersFunctionnormalize

Apply min-max normalization.

core/wing.py:437

↓ 1 callersFunctionnp2tensor

Converts numpy array to torch tensor.

core/wing.py:285

↓ 1 callersFunctionpad_mirror

(img, landmarks)

core/wing.py:389

↓ 1 callersFunctionpreprocess

Preprocess 98-dimensional heatmaps.

core/wing.py:494

↓ 1 callersFunctionr1_reg

(d_out, x_in)

core/solver.py:282

↓ 1 callersFunctionrotate90

(vec)

core/wing.py:346

↓ 1 callersMethodsave

(self, step)

core/checkpoint.py:25

↓ 1 callersFunctionsigmoid

(x, w=1)

core/utils.py:151

↓ 1 callersFunctionslide

Returns a sliding reference window. Args: entries: a list containing two reference images, x_prev and x_next, both of wh

core/utils.py:176

↓ 1 callersFunctiontensor2numpy255

Converts torch tensor to numpy array.

core/wing.py:280

↓ 1 callersMethodtrain

(self, loaders)

core/solver.py:79

↓ 1 callersFunctiontranslate_and_reconstruct

(nets, args, x_src, y_src, x_ref, y_ref, filename)

core/utils.py:63

↓ 1 callersFunctiontranslate_using_latent

(nets, args, x_src, y_trg_list, z_trg_list, psi, filename)

core/utils.py:78

↓ 1 callersFunctiontranslate_using_reference

(nets, args, x_src, x_ref, y_ref, filename)

core/utils.py:102

↓ 1 callersFunctiontruncate

Remove small values in heatmaps.

core/wing.py:449

↓ 1 callersFunctionvecs2R

core/data_loader.py:40

Method__getitem__

(self, index)

core/data_loader.py:67

(self, in_channels, out_channels=1)

(self, fname_template, data_parallel=False, **kwargs)

core/checkpoint.py:16

Method__init__

(self, num_modules, depth, num_features, first_one=False)

core/wing.py:50

Method__init__

(self, height=64, width=64, with_r=False, with_boundary=False)

core/wing.py:92

Method__init__

(self, height, width, with_r, with_boundary, in_channels, first_one=False, *args, **kwargs)

core/wing.py:133

Method__init__

(self, in_planes, out_planes)

core/wing.py:152

Method__init__

(self, fname_wing, fname_celeba_mean, output_size)

core/wing.py:291

Method__init__

(self, root, transform=None)

core/data_loader.py:34

Method__init__

(self, root, transform=None)

core/data_loader.py:52

Method__init__

(self, loader, loader_ref=None, latent_dim=16, mode='')

core/data_loader.py:171

Method__init__

(self, dim_in, dim_out, actv=nn.LeakyReLU(0.2), normalize=False, downsample=False)

core/model.py:24

Method__init__

(self, dim_in, dim_out, style_dim=64, w_hpf=0, actv=nn.LeakyReLU(0.2), upsample=False)

core/model.py:81

Method__init__

(self, w_hpf, device)

core/model.py:124

Method__init__

(self, img_size=256, style_dim=64, max_conv_dim=512, w_hpf=1)

core/model.py:137

Method__init__

(self, latent_dim=16, style_dim=64, num_domains=2)

core/model.py:191

Method__init__

(self, img_size=256, style_dim=64, num_domains=2, max_conv_dim=512)

core/model.py:223

Method__init__

(self, img_size=256, num_domains=2, max_conv_dim=512)

core/model.py:257

Method__len__

(self)

core/data_loader.py:47

Method__len__

(self)

core/data_loader.py:77

next →1–100 of 126, ranked by callers