MCPcopy Create free account

hub / github.com/NVlabs/stylegan2-ada-pytorch / functions

Functions277 in github.com/NVlabs/stylegan2-ada-pytorch

↓ 1 callersFunctionload_network_pkl
(f, force_fp16=False)
legacy.py:20
↓ 1 callersFunctionmain
Train a GAN using the techniques described in the paper "Training Generative Adversarial Networks with Limited Data". Examples: \b #
train.py:438
↓ 1 callersFunctionmake_cache_dir_path
(*paths: str)
dnnlib/util.py:124
↓ 1 callersFunctionmake_transform
( transform: Optional[str], output_width: Optional[int], output_height: Optional[int], resize_
dataset_tool.py:199
↓ 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 callersFunctionopen_cifar10
(tarball: str, *, max_images: Optional[int])
dataset_tool.py:137
↓ 1 callersFunctionopen_dataset
(source, *, max_images: Optional[int])
dataset_tool.py:252
↓ 1 callersFunctionopen_dest
(dest: str)
dataset_tool.py:272
↓ 1 callersFunctionopen_image_folder
(source_dir, *, max_images: Optional[int])
dataset_tool.py:52
↓ 1 callersFunctionopen_image_zip
(source, *, max_images: Optional[int])
dataset_tool.py:80
↓ 1 callersFunctionopen_lmdb
(lmdb_dir: str, *, max_images: Optional[int])
dataset_tool.py:109
↓ 1 callersFunctionopen_mnist
(images_gz: str, *, max_images: Optional[int])
dataset_tool.py:169
↓ 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 callersFunctionproject
( G, target: torch.Tensor, # [C,H,W] and dynamic range [0,255], W & H must match G output resolution
projector.py:25
↓ 1 callersFunctionrecurse
(prefix, tf_net)
legacy.py:78
↓ 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)
training/augment.py:83
↓ 1 callersFunctionrotate3d
(v, theta, **kwargs)
training/augment.py:90
↓ 1 callersFunctionrun_generator
(z, c)
metrics/metric_utils.py:242
↓ 1 callersFunctionrun_projection
Project given image to the latent space of pretrained network pickle. Examples: \b python projector.py --outdir=out --target=~/mytargeti
projector.py:142
↓ 1 callersFunctionscale3d
(sx, sy, sz, **kwargs)
training/augment.py:75
↓ 1 callersMethodset_num_features
(self, num_features)
metrics/metric_utils.py:66
↓ 1 callersFunctionsetup_snapshot_image_grid
(training_set, random_seed=0)
training/training_loop.py:29
↓ 1 callersFunctionsetup_training_loop_kwargs
( # General options (not included in desc). gpus = None, # Number of GPUs: <int>, default = 1 gp
train.py:32
↓ 1 callersFunctionsubprocess_fn
(rank, args, temp_dir)
train.py:363
↓ 1 callersFunctionsubprocess_fn
(rank, args, temp_dir)
calc_metrics.py:28
↓ 1 callersFunctiontranslate3d
(tx, ty, tz, **kwargs)
training/augment.py:60
Method__del__
(self)
training/dataset.py:76
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__
r"""Convenience getter. `collector[name]` is a synonym for `collector.mean(name)`.
torch_utils/training_stats.py:226
Method__getitem__
(self, idx)
training/dataset.py:85
Method__getstate__
(self)
training/dataset.py:207
Method__init__
(self, dataset, rank=0, num_replicas=1, shuffle=True, seed=0, window_size=0.5)
torch_utils/misc.py:110
Method__init__
(self, *args, **kwargs)
torch_utils/persistence.py:103
Method__init__
(self, regex='.*', keep_previous=True)
torch_utils/training_stats.py:133
Method__init__
(self, G, G_kwargs, epsilon, space, sampling, crop, vgg16)
metrics/perceptual_path_length.py:37
Method__init__
(self, G=None, G_kwargs={}, dataset_kwargs={}, num_gpus=1, rank=0, device=None, progress=None, cache=True)
metrics/metric_utils.py:22
Method__init__
(self, capture_all=False, capture_mean_cov=False, max_items=None)
metrics/metric_utils.py:56
Method__init__
(self, tag=None, num_items=None, flush_interval=1000, verbose=False, progress_fn=None, pfn_lo=0, pfn_hi=1000,
metrics/metric_utils.py:136
Method__init__
(self, device, G_mapping, G_synthesis, D, augment_pipe=None, style_mixing_prob=0.9, r1_gamma=10, pl_batch_shri
training/loss.py:24
Method__init__
(self, path, # Path to directory or zip. resolution = None, # Ensure sp
training/dataset.py:155
Method__init__
(self, in_features, # Number of input features. out_features, # N
training/networks.py:90
Method__init__
(self, in_channels, # Number of input channels. out_channels,
training/networks.py:124
Method__init__
(self, z_dim, # Input latent (Z) dimensionality, 0 = no latent. c_dim,
training/networks.py:175
Method__init__
(self, in_channels, # Number of input channels. out_channels,
training/networks.py:255
Method__init__
(self, in_channels, out_channels, w_dim, kernel_size=1, conv_clamp=None, channels_last=False)
training/networks.py:311
Method__init__
(self, in_channels, # Number of input channels, 0 = first block. out_ch
training/networks.py:330
Method__init__
(self, w_dim, # Intermediate latent (W) dimensionality. img_resolution,
training/networks.py:425
Method__init__
(self, in_channels, # Number of input channels, 0 = first block. tmp_ch
training/networks.py:506
Method__init__
(self, group_size, num_channels=1)
training/networks.py:590
Method__init__
(self, in_channels, # Number of input channels. cmap_dim,
training/networks.py:616
Method__init__
(self, c_dim, # Conditioning label (C) dimensionality. img_resolution
training/networks.py:674
Method__init__
(self, xflip=0, rotate90=0, xint=0, xint_max=0.125, scale=0, rotate=0, aniso=0, xfrac=0, scale
training/augment.py:118
Method__init__
(self, file_name: str = None, file_mode: str = "w", should_flush: bool = True)
dnnlib/util.py:59
Method__iter__
(self)
torch_utils/misc.py:123
Method__len__
(self)
training/dataset.py:82
Method__setattr__
(self, name: str, value: Any)
dnnlib/util.py:49
Method_load_raw_labels
(self)
training/dataset.py:222
Function_reconstruct_persistent_obj
r"""Hook that is called internally by the `pickle` module to unpickle a persistent object.
torch_utils/persistence.py:179
Methodaccumulate_gradients
(self, phase, real_img, real_c, gen_z, gen_c, sync, gain)
training/loss.py:57
Functionask_yes_no
Ask the user the question until the user inputs a valid answer.
dnnlib/util.py:153
Functionassert_shape
(tensor, ref_shape)
torch_utils/misc.py:80
Methodbackward
(ctx, grad_output)
torch_utils/ops/grid_sample_gradfix.py:54
Methodbackward
(ctx, grad2_grad_input, grad2_grad_grid)
torch_utils/ops/grid_sample_gradfix.py:70
Methodbackward
(ctx, d_dx)
torch_utils/ops/bias_act.py:189
Methodbackward
(ctx, dy)
torch_utils/ops/upfirdn2d.py:246
Methodbackward
(ctx, dout)
torch_utils/ops/fma.py:29
Methodbackward
(ctx, grad_output)
torch_utils/ops/conv2d_gradfix.py:119
Methodbackward
(ctx, grad2_grad_weight)
torch_utils/ops/conv2d_gradfix.py:151
Functionbias_act
r"""Fused bias and activation function. Adds bias `b` to activation tensor `x`, evaluates activation function `act`, and scales the result by
torch_utils/ops/bias_act.py:55
Functioncalc_metric
(metric, **kwargs)
metrics/metric_main.py:39
Functioncenter_crop
(width, height, img)
dataset_tool.py:217
Functioncenter_crop_wide
(width, height, img)
dataset_tool.py:224
Functioncheck_ddp_consistency
(module, ignore_regex=None)
torch_utils/misc.py:178
Methodclose
(self)
training/dataset.py:200
Functioncompute_feature_stats_for_dataset
(opts, detector_url, detector_kwargs, rel_lo=0, rel_hi=1, batch_size=64, data_loader_kwargs=None, max_items=No
metrics/metric_utils.py:180
Functioncompute_feature_stats_for_generator
(opts, detector_url, detector_kwargs, rel_lo=0, rel_hi=1, batch_size=64, batch_gen=None, jit=False, **stats_kw
metrics/metric_utils.py:232
Functioncompute_fid
(opts, max_real, num_gen)
metrics/frechet_inception_distance.py:20
Functioncompute_is
(opts, num_gen, num_splits)
metrics/inception_score.py:18
Functioncompute_kid
(opts, max_real, num_gen, num_subsets, max_subset_size)
metrics/kernel_inception_distance.py:18
Functioncompute_ppl
(opts, num_samples, epsilon, space, sampling, crop, batch_size, jit=False)
metrics/perceptual_path_length.py:95
Functioncompute_pr
(opts, max_real, num_gen, nhood_size, row_batch_size, col_batch_size)
metrics/precision_recall.py:36
Functionconstant
(value, shape=None, dtype=None, device=None, memory_format=None)
torch_utils/misc.py:22
Functionconstruct_class_by_name
Finds the python class with the given name and constructs it with the given arguments.
dnnlib/util.py:287
Functionconv2d
(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)
torch_utils/ops/conv2d_gradfix.py:35
Functionconv2d_resample
r"""2D convolution with optional up/downsampling. Padding is performed only once at the beginning, not between the operations. Args:
torch_utils/ops/conv2d_resample.py:59
Functionconv_transpose2d
(input, weight, bias=None, stride=1, padding=0, output_padding=0, groups=1, dilation=1)
torch_utils/ops/conv2d_gradfix.py:40
Methodconvert
(self, value, param, ctx)
calc_metrics.py:79
Functioncopy_files_and_create_dirs
Takes in a list of tuples of (src, dst) paths and copies files. Will create all necessary directories.
dnnlib/util.py:348
Functioncopy_params_and_buffers
(src_module, dst_module, require_all=False)
torch_utils/misc.py:153
Functionddp_sync
(module, sync)
torch_utils/misc.py:167
Functiondecorator
(*args, **kwargs)
torch_utils/misc.py:99
Functiondownsample2d
r"""Downsample a batch of 2D images using the given 2D FIR filter. By default, the result is padded so that its shape is a fraction of the input.
torch_utils/ops/upfirdn2d.py:347
Functionfid50k
(opts)
metrics/metric_main.py:115
Functionfid50k_full
(opts)
metrics/metric_main.py:83
← previousnext →101–200 of 277, ranked by callers