MCPcopy Create free account

hub / github.com/chi0tzp/WarpedGANSpace / functions

Functions310 in github.com/chi0tzp/WarpedGANSpace

↓ 32 callersMethodcuda
(self, device=None)
models/SNGAN/sn_gen_resnet.py:66
↓ 30 callersMethodupdate
(self, accuracy, classification_loss, regression_loss, total_loss)
lib/aux.py:22
↓ 20 callersMethodcpu
(self)
models/SNGAN/distribution.py:15
↓ 16 callersMethodto
(self, device)
models/SNGAN/distribution.py:19
↓ 15 callersMethod__init__
(self, kernel, pad, upsample_factor=1)
models/StyleGAN2/model.py:68
↓ 15 callersFunctionupdate
(state_dict, new)
models/StyleGAN2/convert_weight.py:94
↓ 12 callersMethod__init__
(self, num_svs, num_itrs, num_outputs, transpose=False, eps=1e-12)
models/BigGAN/layers.py:58
↓ 12 callersFunctiondownload
(src, sha256sum, dest)
download_models.py:30
↓ 12 callersFunctionget_block
(in_channel, depth, num_units, stride=2)
lib/evaluation/archface/arcface.py:110
↓ 7 callersFunction_unsqueeze_ft
add new dementions at the front and the tail
models/BigGAN/sync_batchnorm/batchnorm.py:29
↓ 6 callersFunctionupdate_progress
(msg, total, progress)
lib/aux.py:107
↓ 5 callersFunctionconvert_conv
(vars, source_name, target_name, bias=True, start=0)
models/StyleGAN2/convert_weight.py:42
↓ 5 callersFunctionupdate_stdout
Update stdout by moving cursor up and erasing line for given number of lines. Args: num_lines (int): number of lines
lib/aux.py:121
↓ 4 callersMethod__init__
(self, num_layers, drop_ratio=0.4, mode='ir_se')
lib/evaluation/archface/arcface.py:134
↓ 4 callersMethod__init__
(self)
models/ProgGAN/model.py:66
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
lib/evaluation/celeba_attributes/celeba_attr_predictor.py:153
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
lib/evaluation/hopenet/hopenet.py:34
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
lib/evaluation/hopenet/hopenet.py:93
↓ 4 callersMethodget
(self)
models/BigGAN/sync_batchnorm/comm.py:32
↓ 4 callersMethodget_w
Return batch of w latent codes given a batch of z latent codes. Args: z (torch.Tensor) : Z-space latent code of size [batch_size,
models/gan_load.py:145
↓ 3 callersMethodW_
(self)
models/BigGAN/layers.py:84
↓ 3 callersMethod__init__
(self, block, layers, attr_file, zero_init
lib/evaluation/celeba_attributes/celeba_attr_predictor.py:107
↓ 3 callersMethod__init__
(self, num_modules=1, num_in=3, num_features = 128, num_out=68, return_features=False)
lib/evaluation/au_detector/hourglass.py:116
↓ 3 callersMethod__init__
(self, G)
models/gan_load.py:22
↓ 3 callersFunctionbuild_biggan
(pretrained_gan_weights, target_classes)
models/gan_load.py:84
↓ 3 callersFunctionbuild_proggan
(pretrained_gan_weights)
models/gan_load.py:123
↓ 3 callersFunctionbuild_sngan
(pretrained_gan_weights, gan_type)
models/gan_load.py:31
↓ 3 callersFunctionbuild_stylegan2
(pretrained_gan_weights, resolution, shift_in_w_space=False)
models/gan_load.py:182
↓ 3 callersFunctionconv1x1
1x1 convolution
lib/evaluation/celeba_attributes/celeba_attr_predictor.py:12
↓ 3 callersFunctionconv1x1
1x1 convolution with padding.
lib/evaluation/au_detector/hourglass.py:12
↓ 3 callersFunctionconv3x3
3x3 convolution with padding
lib/evaluation/celeba_attributes/celeba_attr_predictor.py:7
↓ 3 callersFunctionconv3x3
3x3 convolution with padding.
lib/evaluation/au_detector/hourglass.py:7
↓ 3 callersFunctionconvert_dense
(vars, source_name, target_name)
models/StyleGAN2/convert_weight.py:80
↓ 3 callersFunctionconvert_modconv
(vars, source_name, target_name, flip=False)
models/StyleGAN2/convert_weight.py:14
↓ 3 callersFunctioncrop_face
Crop faces from given images for the given bounding boxes and padding.
traverse_attribute_space.py:37
↓ 3 callersMethodflush
(self)
lib/aux.py:34
↓ 3 callersFunctionfused_bn
(x, mean, var, gain=None, bias=None, eps=1e-5)
models/BigGAN/layers.py:170
↓ 3 callersFunctionmake_kernel
(k)
models/StyleGAN2/model.py:18
↓ 3 callersMethodput
(self, result)
models/BigGAN/sync_batchnorm/comm.py:26
↓ 3 callersFunctionsec2dhms
Convert time into days, hours, minutes, and seconds string format. Args: t (float): time in seconds Returns (string): "<days
lib/aux.py:134
↓ 3 callersFunctionupfirdn2d
(input, kernel, up=1, down=1, pad=(0, 0))
models/StyleGAN2/op/upfirdn2d.py:144
↓ 2 callersMethod__init__
(self, block, layers, num_classes=1000)
lib/evaluation/hopenet/hopenet.py:70
↓ 2 callersMethod__init__
(self, target_shape)
models/SNGAN/sn_gen_resnet.py:16
↓ 2 callersMethod__init__
(self, G, D)
models/BigGAN/BigGAN.py:405
↓ 2 callersMethod_check_input_dim
(self, input)
models/BigGAN/sync_batchnorm/batchnorm.py:218
↓ 2 callersMethod_reshape
(z)
models/gan_load.py:116
↓ 2 callersFunction_sum_ft
sum over the first and last dimention
models/BigGAN/sync_batchnorm/batchnorm.py:24
↓ 2 callersFunctionbatch_detect
Inputs: - img_batch: a torch.Tensor of shape (Batch size, Channels, Height, Width)
lib/evaluation/sfd/detect.py:33
↓ 2 callersFunctionconvert_torgb
(vars, source_name, target_name)
models/StyleGAN2/convert_weight.py:59
↓ 2 callersFunctiondetect
(net, img, device)
lib/evaluation/sfd/detect.py:20
↓ 2 callersFunctionexecute_replication_callbacks
Execute an replication callback `__data_parallel_replicate__` on each module created by original replication. The callback will be invoked w
models/BigGAN/sync_batchnorm/replicate.py:27
↓ 2 callersMethodextract_feats
(self, x)
lib/evaluation/archface/arcface.py:16
↓ 2 callersFunctionfill_statedict
(state_dict, vars, size)
models/StyleGAN2/convert_weight.py:126
↓ 2 callersFunctionfused_leaky_relu
(input, bias, negative_slope=0.2, scale=2 ** 0.5)
models/StyleGAN2/op/fused_act.py:85
↓ 2 callersMethodget_latent
(self, input)
models/StyleGAN2/model.py:356
↓ 2 callersMethodget_means
(self)
lib/aux.py:28
↓ 2 callersFunctiongram_schmidt
(x, ys)
models/BigGAN/layers.py:18
↓ 2 callersFunctionnms
(dets, thresh)
lib/evaluation/sfd/bbox.py:44
↓ 2 callersFunctionsample_z
Sample a random latent code from multi-variate standard Gaussian distribution with/without truncation. Args: batch_size (int) : batch s
lib/aux.py:39
↓ 2 callersFunctionsave_results
Save interpretable paths results and create summarizing GIFs. Args: attributes (list) : list of attribute names att
rank_interpretable_paths.py:97
↓ 1 callersMethod__data_parallel_replicate__
(self, ctx, copy_id)
models/BigGAN/sync_batchnorm/batchnorm.py:110
↓ 1 callersMethod__init__
(self)
lib/evaluation/sfd/net_s3fd.py:23
↓ 1 callersMethod__init__
(self, dim, device='cuda')
models/SNGAN/distribution.py:6
↓ 1 callersMethod__init__
Args: master_callback: a callback to be invoked after having collected messages from slave devices.
models/BigGAN/sync_batchnorm/comm.py:67
↓ 1 callersMethod_compute_mean_std
Compute the mean and standard-deviation with sum and square-sum. This method also maintains the moving average on the master device.
models/BigGAN/sync_batchnorm/batchnorm.py:147
↓ 1 callersMethod_forward
(self, level, inp)
lib/evaluation/au_detector/hourglass.py:89
↓ 1 callersMethod_generate_network
(self, level)
lib/evaluation/au_detector/hourglass.py:77
↓ 1 callersMethodbackward
(ctx, grad_output)
models/StyleGAN2/op/upfirdn2d.py:126
↓ 1 callersFunctionbuild_gan
(gan_type, target_classes, stylegan2_resolution, shift_in_w_space, use_cuda, multi_gpu)
traverse_latent_space.py:44
↓ 1 callersMethodcalculate_pose
(self, face, batch_index, image)
lib/evaluation/hopenet/pose_estimator.py:54
↓ 1 callersFunctionceleba_attr_predictor
(attr_file, pretrained='models/pretrained/celeba_attributes/predictor_1024.pth.tar')
lib/evaluation/celeba_attributes/celeba_attr_predictor.py:187
↓ 1 callersFunctioncreate_exp_dir
Create output directory for current experiment under experiments/wip/ and save given the arguments (json) and the given command (bash script).
lib/aux.py:56
↓ 1 callersFunctioncreate_summarizing_gif
Create a summarizing GIF image given an images root directory (images generated across a certain latent path) and the number of images to appear a
lib/aux.py:178
↓ 1 callersFunctioncreate_summary_md_file
Create summary .md file for the given attributes group. For all given attributes, show the top-k interpretable paths for each latent code (hash).
rank_interpretable_paths.py:181
↓ 1 callersFunctiondecode
Decode locations from predictions using priors to undo the encoding we did for offset regression at train time. Args: loc (tensor):
lib/evaluation/sfd/bbox.py:93
↓ 1 callersMethoddetect_AU
(self, img)
lib/evaluation/au_detector/AU_detector.py:35
↓ 1 callersMethoddetect_from_batch
Detects faces in a given image. This function detects the faces present in a provided BGR(usually) image. The input can be either
lib/evaluation/sfd/core.py:55
↓ 1 callersMethoddetect_from_batch
(self, tensor)
lib/evaluation/sfd/sfd_detector.py:24
↓ 1 callersMethoddetect_from_image
Detects faces in a given image. This function detects the faces present in a provided BGR(usually) image. The input can be either
lib/evaluation/sfd/core.py:33
↓ 1 callersFunctiondiscriminator_arch
(ch=64, attention='64', ksize='333333', dilation='111111')
models/BigGAN/BigGAN.py:247
↓ 1 callersFunctiondiscriminator_fill_statedict
(statedict, vars, size)
models/StyleGAN2/convert_weight.py:105
↓ 1 callersMethodforward_FAN
(self, images)
lib/evaluation/au_detector/AU_detector.py:24
↓ 1 callersFunctiongenerator_arch
(ch=64, attention='64', ksize='333333', dilation='111111')
models/BigGAN/BigGAN.py:13
↓ 1 callersFunctionget_blocks
(num_layers)
lib/evaluation/archface/arcface.py:114
↓ 1 callersFunctionget_concat_h
(img_file_orig, shifted_img_file, size, img_id,
traverse_latent_space.py:79
↓ 1 callersMethodget_starting_iteration
Check if checkpoint file exists (under `self.models_dir`) and set starting iteration at the checkpoint iteration; also load checkpoint weights
lib/trainer.py:74
↓ 1 callersFunctionget_wh
Get width and height of images in given list of paths. Images are expected to have the same resolution. Args: img_paths (list): list of i
lib/aux.py:154
↓ 1 callersFunctiongroupnorm
(x, norm_style)
models/BigGAN/layers.py:256
↓ 1 callersMethodimage2tensor
(image_file)
lib/data.py:21
↓ 1 callersFunctioninit_pretrained_weights
Initialize model with pretrained weights. Layers that don't match with pretrained layers in name or size are kept unchanged.
lib/evaluation/celeba_attributes/celeba_attr_predictor.py:194
↓ 1 callersFunctioninit_weights
(net, init_type='normal', gain=0.02)
lib/evaluation/au_detector/hourglass.py:185
↓ 1 callersMethodinit_weights
(self)
models/BigGAN/BigGAN.py:201
↓ 1 callersMethodinit_weights
(self)
models/BigGAN/BigGAN.py:369
↓ 1 callersFunctionl1
Perform L1-normalization.
rank_interpretable_paths.py:88
↓ 1 callersFunctionl2_norm
(x, axis=1)
lib/evaluation/archface/arcface.py:36
↓ 1 callersMethodlog_progress
Log progress in terms of batch accuracy, classification and regression losses and ETA. Args: iteration (int) : current ite
lib/trainer.py:91
↓ 1 callersFunctionmain
WarpedGANSpace -- Training script. Options: ===[ Pre-trained GAN Generator (G) ]=========================================================
train.py:7
↓ 1 callersFunctionmain
A script for traversing the attribute space of the generated paths (as calculated by `traverse_latent_space.py`). Attribute space includes the fol
traverse_attribute_space.py:66
↓ 1 callersFunctionmain
An auxiliary script for converting a checkpoint file (`checkpoint.pt`) into a support sets (`support_sets.pt`) and a reconstructor (`reconstructor
checkpoint2model.py:6
↓ 1 callersFunctionmain
Download pre-trained attribute detectors models: -- GAN generators -- Spectral Norm GAN [1] for MNIST [2] and AnimeFaces [3] datas
download_models.py:54
next →1–100 of 310, ranked by callers