MCPcopy Create free account

hub / github.com/OpenGVLab/DragGAN / functions

Functions447 in github.com/OpenGVLab/DragGAN

↓ 1 callersFunction_sync
r"""Synchronize the global cumulative counters across devices and processes. Called internally by `Collector.update()`.
draggan/stylegan2/torch_utils/training_stats.py:234
↓ 1 callersFunction_upfirdn2d_ref
Slow reference implementation of `upfirdn2d()` using standard PyTorch ops.
draggan/stylegan2/torch_utils/ops/upfirdn2d.py:169
↓ 1 callersMethodaccumulate_gradients
(self, phase, real_img, real_c, gen_z, gen_c, gain, cur_nimg)
draggan/stylegan2/training/loss.py:20
↓ 1 callersMethodas_dict
r"""Returns the averages accumulated between the last two calls to `update()` as an `dnnlib.EasyDict`. The contents are as follows:
draggan/stylegan2/torch_utils/training_stats.py:212
↓ 1 callersMethodbackward_train
(self)
draggan/deprecated/stylegan2/lpips/dist_model.py:190
↓ 1 callersFunctioncall_func_by_name
Finds the python object with the given name and calls it as a function.
draggan/stylegan2/dnnlib/util.py:279
↓ 1 callersMethodclamp_weights
(self)
draggan/deprecated/stylegan2/lpips/dist_model.py:155
↓ 1 callersMethodclose
Flush, close possible files, and remove stdout/stderr mirroring.
draggan/stylegan2/dnnlib/util.py:100
↓ 1 callersMethodcompute_accuracy
d0, d1 are Variables, judge is a Tensor
draggan/deprecated/stylegan2/lpips/dist_model.py:193
↓ 1 callersFunctionconvert_network_pickle
Convert legacy network pickle into the native PyTorch format. The tool is able to load the main network configurations exported using the TensorF
draggan/stylegan2/legacy.py:294
↓ 1 callersFunctionconvert_tf_discriminator
(tf_D)
draggan/stylegan2/legacy.py:207
↓ 1 callersFunctiondownload_url
(url, output_path)
draggan/utils.py:30
↓ 1 callersFunctiondownload_url
(url, output_path)
draggan/deprecated/utils.py:30
↓ 1 callersFunctiondownload_url
(url, output_path)
draggan/deprecated/stylegan2/lpips/dist_model.py:47
↓ 1 callersFunctiondrag_gan
( g_ema, latent: torch.Tensor, noise, F, handle_points, target_points, mask, m
draggan/deprecated/api.py:103
↓ 1 callersMethodforward_train
(self)
draggan/deprecated/stylegan2/lpips/dist_model.py:176
↓ 1 callersFunctionfused_leaky_relu_native
(input, bias, negative_slope=0.2, scale=2 ** 0.5)
draggan/deprecated/stylegan2/op/fused_act.py:156
↓ 1 callersMethodget_details
(self, idx)
draggan/stylegan2/training/dataset.py:105
↓ 1 callersFunctionget_lr
(t, initial_lr, rampdown=0.25, rampup=0.05)
draggan/deprecated/stylegan2/inversion.py:46
↓ 1 callersFunctionget_obj_by_name
Finds the python object with the given name.
draggan/stylegan2/dnnlib/util.py:273
↓ 1 callersFunctionget_path
(base_path)
draggan/deprecated/stylegan2/lpips/dist_model.py:33
↓ 1 callersFunctionhook
(module, input, output)
draggan/draggan.py:48
↓ 1 callersFunctionimresize
(image, size)
draggan/stylegan2/training/networks_stylegan3.py:537
↓ 1 callersFunctionimresize
(image, size)
draggan/stylegan2/training/networks.py:576
↓ 1 callersMethodinitialize
INPUTS model - ['net-lin'] for linearly calibrated network ['net'] for off-the-shelf network
draggan/deprecated/stylegan2/lpips/dist_model.py:57
↓ 1 callersFunctionis_top_level_function
Determine whether the given object is a top-level function, i.e., defined at module scope using 'def'.
draggan/stylegan2/dnnlib/util.py:298
↓ 1 callersFunctionis_url
Determine whether the given object is a valid URL string.
draggan/stylegan2/dnnlib/util.py:364
↓ 1 callersFunctionlatent_noise
(latent, strength)
draggan/deprecated/stylegan2/inversion.py:54
↓ 1 callersFunctionload_model
Loads a pretrained StyleGAN2-ADA generator network from a pickle file. Args: network_pkl (str): The URL or local path to the network
draggan/draggan.py:20
↓ 1 callersFunctionload_network_pkl
(f, force_fp16=False)
draggan/stylegan2/legacy.py:20
↓ 1 callersFunctionmain
()
gradio_app.py:204
↓ 1 callersFunctionmain
()
draggan/web.py:205
↓ 1 callersFunctionmain
()
draggan/deprecated/web.py:207
↓ 1 callersFunctionmake_cache_dir_path
(*paths: str)
draggan/stylegan2/dnnlib/util.py:124
↓ 1 callersFunctionmake_image
(tensor)
draggan/deprecated/stylegan2/inversion.py:60
↓ 1 callersMethodmake_noise
(self)
draggan/deprecated/stylegan2/model.py:493
↓ 1 callersFunctionmodulated_conv2d
( x, # Input tensor: [batch_size, in_channels, in_height, in_width] w,
draggan/stylegan2/training/networks_stylegan3.py:26
↓ 1 callersFunctionmotion_supervison
(handle_points, target_points, F, r1, device)
draggan/draggan.py:269
↓ 1 callersFunctionmotion_supervison
(handle_points, target_points, F2, r1, device)
draggan/deprecated/api.py:158
↓ 1 callersMethodname
(self)
draggan/deprecated/stylegan2/lpips/dist_model.py:54
↓ 1 callersFunctionnoise_normalize_
(noises)
draggan/deprecated/stylegan2/inversion.py:38
↓ 1 callersFunctionnoise_regularize
(noises)
draggan/deprecated/stylegan2/inversion.py:15
↓ 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
draggan/stylegan2/torch_utils/training_stats.py:180
↓ 1 callersFunctionpersistent_class
r"""Class decorator that extends a given class to save its source code when pickled. Example: from torch_utils import persistence
draggan/stylegan2/torch_utils/persistence.py:35
↓ 1 callersFunctionpoint_tracking
( F: torch.Tensor, F0: torch.Tensor, handle_points: torch.Tensor, handle_points0: torch.Tensor
draggan/draggan.py:322
↓ 1 callersFunctionpoint_tracking
( F: torch.Tensor, F0: torch.Tensor, handle_points: torch.Tensor, handle_points0: torch.Tensor
draggan/deprecated/api.py:211
↓ 1 callersFunctionrecurse
(prefix, tf_net)
draggan/stylegan2/legacy.py:78
↓ 1 callersFunctionrecurse
(obj)
draggan/stylegan2/torch_utils/persistence.py:236
↓ 1 callersFunctionregister_hook
(G)
draggan/draggan.py:60
↓ 1 callersFunctionreport
r"""Broadcasts the given set of scalars to all interested instances of `Collector`, across device and process boundaries. This function is ex
draggan/stylegan2/torch_utils/training_stats.py:56
↓ 1 callersFunctionrotate2d
(theta, **kwargs)
draggan/stylegan2/training/augment.py:88
↓ 1 callersFunctionrotate3d
(v, theta, **kwargs)
draggan/stylegan2/training/augment.py:95
↓ 1 callersFunctionscale3d
(sx, sy, sz, **kwargs)
draggan/stylegan2/training/augment.py:80
↓ 1 callersFunctionsetup_snapshot_image_grid
(training_set, random_seed=0)
draggan/stylegan2/training/training_loop.py:31
↓ 1 callersFunctionstylegan2
( size=1024, channel_multiplier=2, latent=512, n_mlp=8, ckpt='stylegan2-ffhq-config-f.pt'
draggan/deprecated/api.py:88
↓ 1 callersFunctiontensor2im
(image_tensor, imtype=np.uint8, cent=1., factor=255./2.)
draggan/deprecated/stylegan2/lpips/util.py:106
↓ 1 callersFunctiontensor2np
(tensor_obj)
draggan/deprecated/stylegan2/lpips/util.py:62
↓ 1 callersFunctiontranslate3d
(tx, ty, tz, **kwargs)
draggan/stylegan2/training/augment.py:65
Method__del__
(self)
draggan/stylegan2/training/dataset.py:78
Method__delattr__
(self, name: str)
draggan/stylegan2/dnnlib/util.py:52
Method__enter__
(self)
draggan/stylegan2/torch_utils/misc.py:70
Method__enter__
(self)
draggan/stylegan2/dnnlib/util.py:72
Method__exit__
(self, exc_type: Any, exc_value: Any, traceback: Any)
draggan/stylegan2/dnnlib/util.py:75
Method__getattr__
(self, name: str)
draggan/stylegan2/dnnlib/util.py:43
Method__getitem__
r"""Convenience getter. `collector[name]` is a synonym for `collector.mean(name)`.
draggan/stylegan2/torch_utils/training_stats.py:226
Method__getitem__
(self, idx)
draggan/stylegan2/training/dataset.py:87
Method__getstate__
(self)
draggan/stylegan2/training/dataset.py:209
Method__init__
(self, **kwargs)
draggan/deprecated/web.py:48
Method__init__
(self)
draggan/deprecated/stylegan2/model.py:30
Method__init__
(self, kernel, factor=2)
draggan/deprecated/stylegan2/model.py:49
Method__init__
(self, kernel, factor=2)
draggan/deprecated/stylegan2/model.py:70
Method__init__
( self, in_channel, out_channel, kernel_size, stride=1, padding=0, bias=True )
draggan/deprecated/stylegan2/model.py:110
Method__init__
( self, in_dim, out_dim, bias=True, bias_init=0, lr_mul=1, activation=None )
draggan/deprecated/stylegan2/model.py:148
Method__init__
( self, in_channel, out_channel, kernel_size, style_dim, demod
draggan/deprecated/stylegan2/model.py:185
Method__init__
(self)
draggan/deprecated/stylegan2/model.py:321
Method__init__
(self, channel, size_in=4, size_out=4)
draggan/deprecated/stylegan2/model.py:335
Method__init__
( self, in_channel, out_channel, kernel_size, style_dim, upsam
draggan/deprecated/stylegan2/model.py:348
Method__init__
(self, in_channel, style_dim, upsample=True, blur_kernel=[1, 3, 3, 1])
draggan/deprecated/stylegan2/model.py:385
Method__init__
( self, size, style_dim, n_mlp, channel_multiplier=2, blur_ker
draggan/deprecated/stylegan2/model.py:407
Method__init__
( self, in_channel, out_channel, kernel_size, downsample=False,
draggan/deprecated/stylegan2/model.py:590
Method__init__
(self, in_channel, out_channel, blur_kernel=[1, 3, 3, 1])
draggan/deprecated/stylegan2/model.py:635
Method__init__
(self, size, channel_multiplier=2, blur_kernel=[1, 3, 3, 1])
draggan/deprecated/stylegan2/model.py:656
Method__init__
(self, channel, bias=True, negative_slope=0.2, scale=2 ** 0.5)
draggan/deprecated/stylegan2/op/fused_act.py:140
Method__init__
(self)
draggan/deprecated/stylegan2/lpips/base_model.py:9
Method__init__
(self, requires_grad=False, pretrained=True)
draggan/deprecated/stylegan2/lpips/pretrained_networks.py:7
Method__init__
(self, requires_grad=False, pretrained=True)
draggan/deprecated/stylegan2/lpips/pretrained_networks.py:58
Method__init__
(self, requires_grad=False, pretrained=True, num=18)
draggan/deprecated/stylegan2/lpips/pretrained_networks.py:140
Method__init__
(self, model='net-lin', net='alex', colorspace='rgb', spatial=False, use_gpu=True, gpu_ids=[0])
draggan/deprecated/stylegan2/lpips/util.py:14
Method__init__
(self)
draggan/deprecated/stylegan2/lpips/networks_basic.py:96
Method__init__
(self, chn_in, chn_out=1, use_dropout=False)
draggan/deprecated/stylegan2/lpips/networks_basic.py:107
Method__init__
(self, chn_mid=32, use_sigmoid=True)
draggan/deprecated/stylegan2/lpips/networks_basic.py:117
Method__init__
(self, chn_mid=32)
draggan/deprecated/stylegan2/lpips/networks_basic.py:133
Method__init__
(self, use_gpu=True, colorspace='Lab')
draggan/deprecated/stylegan2/lpips/networks_basic.py:146
Method__init__
(self, dataset, rank=0, num_replicas=1, shuffle=True, seed=0, window_size=0.5)
draggan/stylegan2/torch_utils/misc.py:110
Method__init__
(self, *args, **kwargs)
draggan/stylegan2/torch_utils/persistence.py:103
Method__init__
(self, regex='.*', keep_previous=True)
draggan/stylegan2/torch_utils/training_stats.py:133
Method__init__
(self, device, G, D, augment_pipe=None, r1_gamma=10, style_mixing_prob=0, pl_weight=0, pl_batch_shrink=2, pl_d
draggan/stylegan2/training/loss.py:26
Method__init__
(self, in_features, # Number of input features. out_features, # N
draggan/stylegan2/training/networks_stylegan3.py:71
Method__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
draggan/stylegan2/training/networks_stylegan3.py:111
Method__init__
(self, w_dim, # Intermediate latent (W) dimensionality. channels, # Number of o
draggan/stylegan2/training/networks_stylegan3.py:171
← previousnext →101–200 of 447, ranked by callers