MCPcopy Create free account

hub / github.com/VITA-Group/TransGAN / functions

Functions560 in github.com/VITA-Group/TransGAN

↓ 1 callersFunctioncalculate_activation_statistics
Calculation of the statistics used by the FID. Params: -- gen_imgs : gen_imgs, tensor -- model : Instance of inception model
utils/torch_fid_score.py:212
↓ 1 callersFunctioncalculate_activation_statistics_from_files
Calculation of the statistics used by the FID. Params: -- files : list of paths to image files. Images need to have same dimensions for a
utils/fid_score.py:248
↓ 1 callersFunctioncalculate_fid_given_paths
Calculates the FID of two paths.
utils/fid_score.py:313
↓ 1 callersFunctioncalculate_fid_given_paths_torch
Calculates the FID of two paths :param gen_imgs: The value range of gen_imgs should be (-1, 1). Just the output of tanh. :param path: fid
utils/torch_fid_score.py:255
↓ 1 callersFunctioncalculate_frechet_distance
Numpy implementation of the Frechet Distance. The Frechet distance between two multivariate Gaussians X_1 ~ N(mu_1, C_1) and X_2 ~ N(mu_2, C_2
utils/fid_score.py:116
↓ 1 callersFunctioncall_func_by_name
Finds the python object with the given name and calls it as a function.
dnnlib/util.py:279
↓ 1 callersFunctioncheck_or_download_inception
Checks if the path to the inception file is valid, or downloads the file if it is not present.
utils/fid_score.py:277
↓ 1 callersFunctioncreate_inception_graph
Creates a graph from saved GraphDef file.
utils/fid_score.py:40
↓ 1 callersFunctioncur_stages
Return current stage. :param epoch: current epoch. :return: current stage
functions.py:26
↓ 1 callersFunctiondrop_path
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks). This is the same as the DropConnect impl I created for Ef
models_search/ViT_helper.py:4
↓ 1 callersFunctionfid_inception_v3
Build pretrained Inception model for FID computation The Inception model for FID computation uses a different set of weights and has a slightl
utils/inception_model.py:163
↓ 1 callersFunctionfid_inception_v3
Build pretrained Inception model for FID computation The Inception model for FID computation uses a different set of weights and has a slightl
utils/inception.py:163
↓ 1 callersMethodforward_features
(self, x, aug=True, epoch=400)
models_search/ViT_custom_scale2.py:497
↓ 1 callersMethodforward_features
(self, x)
models_search/ViT_custom_rp.py:407
↓ 1 callersMethodforward_features
(self, x, aug=True, epoch=400)
models_search/Celeba256_dis.py:542
↓ 1 callersMethodforward_features
(self, x)
models_search/ViT_scale3_local_new_rp.py:515
↓ 1 callersMethodforward_features
(self, x, aug=True, epoch=400)
models_search/ViT_custom_scale2_rp_noise.py:504
↓ 1 callersMethodforward_features
(self, x)
models_search/ViT_custom_local544444_256_rp_noise.py:560
↓ 1 callersMethodforward_features
(self, x)
models_search/ViT_custom_local544444_256_rp.py:553
↓ 1 callersMethodforward_features
(self, x)
models_search/Celeba256_gen.py:637
↓ 1 callersMethodforward_features
(self, x)
models_search/ViT_custom.py:383
↓ 1 callersFunctionget_activations
Calculates the activations of the pool_3 layer for all images. Params: -- images : Numpy array of dimension (n_images, hi, wi, 3). The v
utils/fid_score.py:77
↓ 1 callersFunctionget_activations
Calculates the activations of the pool_3 layer for all images. Params: -- files : List of image files paths -- model : Instanc
utils/torch_fid_score.py:114
↓ 1 callersFunctionget_activations_from_files
Calculates the activations of the pool_3 layer for all images. Params: -- files : list of paths to image files. Images need to have same
utils/fid_score.py:212
↓ 1 callersFunctionget_fid
(args, fid_stat, epoch, gen_net, num_img, gen_batch_size, val_batch_size, writer_dict=None, cls_idx=None)
utils/torch_fid_score.py:291
↓ 1 callersFunctionget_inception_score
(images, splits=10)
utils/inception_score.py:31
↓ 1 callersFunctionget_is
Get inception score. :param args: :param gen_net: :param num_img: :return: Inception score
functions.py:247
↓ 1 callersFunctionget_obj_by_name
Finds the python object with the given name.
dnnlib/util.py:273
↓ 1 callersMethodget_var
(self)
utils/utils.py:238
↓ 1 callersFunctionis_top_level_function
Determine whether the given object is a top-level function, i.e., defined at module scope using 'def'.
dnnlib/util.py:298
↓ 1 callersFunctionis_url
Determine whether the given object is a valid URL string.
dnnlib/util.py:364
↓ 1 callersFunctionload_image_batch
Convenience method for batch-loading images Params: -- files : list of paths to image files. Images need to have same dimensions for all fi
utils/fid_score.py:202
↓ 1 callersFunctionmain
()
flops.py:34
↓ 1 callersFunctionmain
()
test.py:83
↓ 1 callersFunctionmain
()
train_derived.py:30
↓ 1 callersFunctionmain
()
utils/cal_fid_stat.py:36
↓ 1 callersFunctionmain_worker
(gpu, ngpus_per_node, args)
train_derived.py:67
↓ 1 callersFunctionmake_cache_dir_path
(*paths: str)
dnnlib/util.py:124
↓ 1 callersFunctionmake_grid
Make a grid of images. Args: tensor (Tensor or list): 4D mini-batch Tensor of shape (B x C x H x W) or a list of images a
utils/utils.py:26
↓ 1 callersMethodnum
r"""Returns the number of scalars that were accumulated for the given statistic between the last two calls to `update()`, or zero if n
torch_utils/training_stats.py:180
↓ 1 callersFunctionparse_args
()
utils/cal_fid_stat.py:18
↓ 1 callersFunctionparse_args
()
exps/celeba_hq_256_test.py:6
↓ 1 callersFunctionparse_args
()
exps/cifar_train.py:6
↓ 1 callersFunctionparse_args
()
exps/cifar_test.py:6
↓ 1 callersFunctionparse_args
()
exps/celeba_hq_256_train.py:6
↓ 1 callersFunctionparse_args
()
exps/church_256_train.py:6
↓ 1 callersFunctionparse_args
()
exps/stl_train.py:6
↓ 1 callersFunctionpersistent_class
r"""Class decorator that extends a given class to save its source code when pickled. Example: from torch_utils import persistence
torch_utils/persistence.py:35
↓ 1 callersFunctionpixel_upsample
(x, H, W)
models_search/ViT_custom_rp.py:187
↓ 1 callersFunctionpixel_upsample
(x, H, W)
models_search/ViT_custom.py:166
↓ 1 callersFunctionrand_bbox
(size, lam)
models_search/diff_aug.py:505
↓ 1 callersFunctionrecurse
(obj)
torch_utils/persistence.py:236
↓ 1 callersFunctionreport
r"""Broadcasts the given set of scalars to all interested instances of `Collector`, across device and process boundaries. This function is ex
torch_utils/training_stats.py:56
↓ 1 callersFunctionrotate2d
(theta, **kwargs)
models_search/ada.py:67
↓ 1 callersFunctionrotate3d
(v, theta, **kwargs)
models_search/ada.py:74
↓ 1 callersFunctionsave_checkpoint
(states, is_best, output_dir, filename='checkpoint.pth')
utils/utils.py:196
↓ 1 callersFunctionsave_image
Save a given Tensor into an image file. Args: tensor (Tensor or list): Image to be saved. If given a mini-batch tensor, s
utils/utils.py:128
↓ 1 callersFunctionsave_samples
(args, fixed_z, fid_stat, epoch, gen_net: nn.Module, writer_dict, clean_dir=True)
functions.py:341
↓ 1 callersFunctionsqrt_newton_schulz
(A, numIters, dtype=None)
utils/torch_fid_score.py:65
↓ 1 callersFunctiontorch_calculate_frechet_distance
Pytorch implementation of the Frechet Distance. Taken from https://github.com/bioinf-jku/TTUR The Frechet distance between two multivariate Ga
utils/torch_fid_score.py:179
↓ 1 callersFunctiontorch_cov
Estimate a covariance matrix given data. Covariance indicates the level to which two variables vary together. If we examine N-dimensional samp
utils/torch_fid_score.py:84
↓ 1 callersFunctiontrain
(args, gen_net: nn.Module, dis_net: nn.Module, gen_optimizer, dis_optimizer, gen_avg_param, train_loader,
functions.py:70
↓ 1 callersFunctionwindow_partition
Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
models_search/ViT_scale3_local_new_rp.py:169
↓ 1 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
models_search/ViT_scale3_local_new_rp.py:183
FunctionPYBIND11_MODULE
torch_utils/ops/bias_act.cpp:94
FunctionPYBIND11_MODULE
torch_utils/ops/upfirdn2d.cpp:98
Method__delattr__
(self, name: str)
dnnlib/util.py:52
Method__enter__
(self)
torch_utils/misc.py:70
Method__enter__
(self)
dnnlib/util.py:72
Method__exit__
(self, exc_type: Any, exc_value: Any, traceback: Any)
dnnlib/util.py:75
Method__getattr__
(self, name: str)
dnnlib/util.py:43
Method__getitem__
obtain the image (read and transform) :param idx: index of the file required :return: img => image array
celeba.py:57
Method__getitem__
obtain the image (read and transform) :param idx: index of the file required :return: img => image array
celeba.py:122
Method__getitem__
r"""Convenience getter. `collector[name]` is a synonym for `collector.mean(name)`.
torch_utils/training_stats.py:226
Method__init__
constructor for the class :param data_dir: path to the directory containing the data :param transform: transforms to be appli
celeba.py:37
Method__init__
constructor for the class :param data_dir: path to the directory containing the data :param transform: transforms to be appli
celeba.py:102
Method__init__
(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8, weight_decay=1e-2, amsgrad=False)
adamw.py:36
Method__init__
(self, args, cur_img_size=None, bs=None)
datasets.py:14
Method__init__
(self, optimizer, start_lr, end_lr, decay_start_step, decay_end_step)
functions.py:398
Method__init__
(self, in_channels, pool_features)
utils/inception_model.py:191
Method__init__
(self, in_channels, channels_7x7)
utils/inception_model.py:217
Method__init__
(self, in_channels)
utils/inception_model.py:246
Method__init__
(self, in_channels)
utils/inception_model.py:280
Method__init__
(self, WIN_SIZE)
utils/utils.py:204
Method__init__
(self, in_channels, pool_features)
utils/inception.py:192
Method__init__
(self, in_channels, channels_7x7)
utils/inception.py:218
Method__init__
(self, in_channels)
utils/inception.py:247
Method__init__
(self, in_channels)
utils/inception.py:281
Method__init__
(self)
models_search/ViT_custom_scale2.py:35
Method__init__
(self, dim)
models_search/ViT_custom_scale2.py:48
Method__init__
(self, act_layer)
models_search/ViT_custom_scale2.py:66
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., window_size=16)
models_search/ViT_custom_scale2.py:96
Method__init__
(self, norm_layer, dim)
models_search/ViT_custom_scale2.py:146
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
models_search/ViT_custom_scale2.py:201
Method__init__
(self, depth, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., drop_path=0.
models_search/ViT_custom_scale2.py:220
Method__init__
(self, args, img_size=224, patch_size=16, in_chans=3, num_classes=10, embed_dim=384, depth=5,
models_search/ViT_custom_scale2.py:254
Method__init__
(self, block_size=2)
models_search/ViT_custom_scale2.py:375
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
models_search/ViT_custom_scale2.py:390
Method__init__
(self, args, img_size=32, patch_size=None, in_chans=3, num_classes=1, embed_dim=None, depth=7,
models_search/ViT_custom_scale2.py:410
Method__init__
(self)
models_search/ViT_custom_rp.py:11
← previousnext →101–200 of 560, ranked by callers