MCPcopy Create free account

hub / github.com/OpenGVLab/DragGAN / functions

Functions447 in github.com/OpenGVLab/DragGAN

↓ 44 callersFunctionkwarg
(tf_name, default=None, none=None)
draggan/stylegan2/legacy.py:116
↓ 39 callersMethodmean
r"""Returns the mean of the scalars that were accumulated for the given statistic between the last two calls to `update()`, or NaN if
draggan/stylegan2/torch_utils/training_stats.py:188
↓ 18 callersMethodupdate
r"""Copies current values of the internal counters to the user-visible state and resets them for the next round. If `keep_previous=Tr
draggan/stylegan2/torch_utils/training_stats.py:147
↓ 14 callersMethod__init__
(self, kernel, pad, upsample_factor=1)
draggan/deprecated/stylegan2/model.py:91
↓ 11 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
draggan/stylegan2/training/networks.py:534
↓ 10 callersFunctionto_image
(tensor)
draggan/deprecated/web.py:52
↓ 7 callersMethodgenerate
( self, latent, noise, )
draggan/deprecated/api.py:64
↓ 7 callersFunctionmatrix
(*rows, device=None)
draggan/stylegan2/training/augment.py:48
↓ 6 callersFunction_conv2d_wrapper
Wrapper for the underlying `conv2d()` and `conv_transpose2d()` implementations.
draggan/stylegan2/torch_utils/ops/conv2d_resample.py:29
↓ 6 callersFunction_parse_padding
(padding)
draggan/stylegan2/torch_utils/ops/upfirdn2d.py:46
↓ 6 callersFunction_parse_scaling
(scaling)
draggan/stylegan2/torch_utils/ops/upfirdn2d.py:37
↓ 5 callersMethod__init__
(self, pnet_type='vgg', pnet_rand=False, pnet_tune=False, use_dropout=True, spatial=False, version='0.1', lpip
draggan/deprecated/stylegan2/lpips/networks_basic.py:29
↓ 5 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
draggan/stylegan2/training/networks_stylegan3.py:497
↓ 5 callersFunction_get_filter_size
(f)
draggan/stylegan2/torch_utils/ops/upfirdn2d.py:57
↓ 5 callersMethodprepare
( self, styles, inject_index=None, truncation=1, truncation_latent=Non
draggan/deprecated/api.py:13
↓ 5 callersFunctionscale2d_inv
(sx, sy, **kwargs)
draggan/stylegan2/training/augment.py:108
↓ 4 callersFunction_conv2d_gradfix
(transpose, weight_shape, stride, padding, output_padding, dilation, groups)
draggan/stylegan2/torch_utils/ops/conv2d_gradfix.py:68
↓ 4 callersMethod_get_raw_labels
(self)
draggan/stylegan2/training/dataset.py:53
↓ 4 callersFunction_tuple_of_ints
(xs, ndim)
draggan/stylegan2/torch_utils/ops/conv2d_gradfix.py:58
↓ 4 callersFunctionadd_points_to_image
(image, points, size=5)
gradio_app.py:55
↓ 4 callersFunctionadd_points_to_image
(image, points, size=5)
draggan/web.py:56
↓ 4 callersFunctionadd_points_to_image
(image, points, size=5)
draggan/deprecated/web.py:60
↓ 4 callersMethodbackward
(ctx, grad_output)
draggan/deprecated/stylegan2/op/upfirdn2d.py:136
↓ 4 callersMethodbackward
(ctx, dy)
draggan/stylegan2/torch_utils/ops/bias_act.py:161
↓ 4 callersFunctionconv2d_gradfix
( transpose, weight_shape, stride, padding, output_padding, dilation, groups )
draggan/deprecated/stylegan2/op/conv2d_gradfix.py:106
↓ 4 callersFunctionensure_tuple
(xs, ndim)
draggan/deprecated/stylegan2/op/conv2d_gradfix.py:97
↓ 4 callersMethodflush
Flush written text to both stdout and a file, if open.
draggan/stylegan2/dnnlib/util.py:93
↓ 4 callersMethodforward
(self, in0, in1, retPerLayer=None)
draggan/deprecated/stylegan2/lpips/networks_basic.py:153
↓ 4 callersFunctionforward_G
Forward pass through the generator network. Args: G (torch.nn.Module): The generator network. W (torch.Tensor): The latent c
draggan/draggan.py:116
↓ 3 callersMethod__init__
(self, requires_grad=False, pretrained=True)
draggan/deprecated/stylegan2/lpips/pretrained_networks.py:98
↓ 3 callersMethod_file_ext
(fname)
draggan/stylegan2/training/dataset.py:186
↓ 3 callersMethod_get_delta
r"""Returns the raw moments that were accumulated for the given statistic between the last two calls to `update()`, or zero if no scal
draggan/stylegan2/torch_utils/training_stats.py:170
↓ 3 callersFunction_unbroadcast
(x, shape)
draggan/stylegan2/torch_utils/ops/fma.py:49
↓ 3 callersMethodforward
Function computes the distance between image patches in0 and in1 INPUTS in0, in1 - torch.Tensor object of shape Nx3xXxY - image p
draggan/deprecated/stylegan2/lpips/dist_model.py:137
↓ 3 callersFunctionget_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
draggan/stylegan2/dnnlib/util.py:263
↓ 3 callersFunctioninverse_image
( g_ema, image, image_size=256, config=InverseConfig() )
draggan/deprecated/stylegan2/inversion.py:87
↓ 3 callersFunctionmake_kernel
(k)
draggan/deprecated/stylegan2/model.py:37
↓ 3 callersFunctionnamed_params_and_buffers
(module)
draggan/stylegan2/torch_utils/misc.py:149
↓ 3 callersFunctionrotate2d_inv
(theta, **kwargs)
draggan/stylegan2/training/augment.py:111
↓ 3 callersMethodrun_D
(self, img, c, blur_sigma=0, update_emas=False)
draggan/stylegan2/training/loss.py:52
↓ 3 callersMethodrun_G
(self, z, c, update_emas=False)
draggan/stylegan2/training/loss.py:42
↓ 3 callersMethodsave
(self, path, label)
draggan/deprecated/stylegan2/lpips/dist_model.py:223
↓ 3 callersFunctionsave_image_grid
(img, fname, drange, grid_size)
draggan/stylegan2/training/training_loop.py:70
↓ 3 callersMethodsave_network
(self, network, path, network_label, epoch_label)
draggan/deprecated/stylegan2/lpips/base_model.py:38
↓ 3 callersFunctionscale2d
(sx, sy, **kwargs)
draggan/stylegan2/training/augment.py:73
↓ 3 callersFunctiontranslate2d
(tx, ty, **kwargs)
draggan/stylegan2/training/augment.py:58
↓ 3 callersFunctiontranslate2d_inv
(tx, ty, **kwargs)
draggan/stylegan2/training/augment.py:105
↓ 3 callersFunctionupfirdn2d
(input, kernel, up=1, down=1, pad=(0, 0))
draggan/deprecated/stylegan2/op/upfirdn2d.py:157
↓ 3 callersFunctionupfirdn2d
r"""Pad, upsample, filter, and downsample a batch of 2D images. Performs the following sequence of operations for each channel: 1. Upsample
draggan/stylegan2/torch_utils/ops/upfirdn2d.py:120
↓ 2 callersMethod__setattr__
(self, name: str, value: Any)
draggan/stylegan2/dnnlib/util.py:49
↓ 2 callersFunction_collect_tf_params
(tf_net)
draggan/stylegan2/legacy.py:75
↓ 2 callersFunction_get_weight_shape
(w)
draggan/stylegan2/torch_utils/ops/conv2d_resample.py:21
↓ 2 callersMethod_get_zipfile
(self)
draggan/stylegan2/training/dataset.py:189
↓ 2 callersMethod_open_file
(self, fname)
draggan/stylegan2/training/dataset.py:195
↓ 2 callersFunction_populate_module_params
(module, *patterns)
draggan/stylegan2/legacy.py:88
↓ 2 callersFunction_should_use_custom_op
(input)
draggan/stylegan2/torch_utils/ops/conv2d_gradfix.py:47
↓ 2 callersFunction_upfirdn2d_cuda
Fast CUDA implementation of `upfirdn2d()` using custom ops.
draggan/stylegan2/torch_utils/ops/upfirdn2d.py:214
↓ 2 callersFunction_upfirdn2d_native
( input, kernel, up_x, up_y, down_x, down_y, pad_x0, pad_x1, pad_y0, pad_y1 )
draggan/deprecated/stylegan2/op/upfirdn2d.py:191
↓ 2 callersFunctioncalc_output_padding
(input_shape, output_shape)
draggan/deprecated/stylegan2/op/conv2d_gradfix.py:124
↓ 2 callersFunctioncalc_output_padding
(input_shape, output_shape)
draggan/stylegan2/torch_utils/ops/conv2d_gradfix.py:95
↓ 2 callersMethodclose
(self)
draggan/stylegan2/training/dataset.py:66
↓ 2 callersFunctionconvert_tf_generator
(tf_G)
draggan/stylegan2/legacy.py:109
↓ 2 callersFunctioncould_use_op
(input)
draggan/deprecated/stylegan2/op/conv2d_gradfix.py:78
↓ 2 callersMethoddesign_lowpass_filter
(numtaps, cutoff, width, fs, radial=False)
draggan/stylegan2/training/networks_stylegan3.py:365
↓ 2 callersFunctiondrag_gan
( W, G, handle_points, target_points, mask, max_iters=1000, r1=3, r2=12, l
draggan/draggan.py:188
↓ 2 callersFunctionfused_leaky_relu
(input, bias=None, negative_slope=0.2, scale=2 ** 0.5)
draggan/deprecated/stylegan2/op/fused_act.py:119
↓ 2 callersFunctionget_ellipse_coords
Returns the coordinates of an ellipse centered at the given point. Args: point (Tuple[int, int]): The center point of the ellipse.
draggan/utils.py:62
↓ 2 callersFunctionget_ellipse_coords
Returns the coordinates of an ellipse centered at the given point. Args: point (Tuple[int, int]): The center point of the ellipse.
draggan/deprecated/utils.py:62
↓ 2 callersMethodget_label
(self, idx)
draggan/stylegan2/training/dataset.py:97
↓ 2 callersFunctionget_module_from_obj_name
Searches for the underlying module behind the name to some python object. Returns the module and the object name (original name with module part r
draggan/stylegan2/dnnlib/util.py:222
↓ 2 callersFunctionim2tensor
(image, imtype=np.uint8, cent=1., factor=255./2.)
draggan/deprecated/stylegan2/lpips/util.py:111
↓ 2 callersFunctionis_persistent
r"""Test whether the given object or class is persistent, i.e., whether it will save its source code when pickled.
draggan/stylegan2/torch_utils/persistence.py:134
↓ 2 callersFunctionmodulated_conv2d
( x, # Input tensor of shape [batch_size, in_channels, in_height, in_width].
draggan/stylegan2/training/networks.py:33
↓ 2 callersMethodnames
r"""Returns the names of all statistics broadcasted so far that match the regular expression specified at construction time.
draggan/stylegan2/torch_utils/training_stats.py:141
↓ 2 callersFunctionnan_to_num
(input, nan=0.0, posinf=None, neginf=None, *, out=None)
draggan/stylegan2/torch_utils/misc.py:49
↓ 2 callersFunctionnormalize_2nd_moment
(x, dim=1, eps=1e-8)
draggan/stylegan2/training/networks.py:27
↓ 2 callersFunctionnp2tensor
(np_obj)
draggan/deprecated/stylegan2/lpips/util.py:66
↓ 2 callersMethodsave
(self, label)
draggan/deprecated/stylegan2/lpips/base_model.py:34
↓ 2 callersFunctionspatial_average
(in_tens, keepdim=True)
draggan/deprecated/stylegan2/lpips/networks_basic.py:18
↓ 2 callersMethodstd
r"""Returns the standard deviation of the scalars that were accumulated for the given statistic between the last two calls to `update(
draggan/stylegan2/torch_utils/training_stats.py:198
↓ 2 callersFunctionto_image
(tensor)
gradio_app.py:47
↓ 2 callersFunctionto_image
(tensor)
draggan/web.py:48
↓ 2 callersFunctionupsample
(in_tens, out_H=64)
draggan/deprecated/stylegan2/lpips/networks_basic.py:21
↓ 2 callersMethodwrite
Write text to stdout (and a file) and optionally flush.
draggan/stylegan2/dnnlib/util.py:78
↓ 1 callersMethod__getstate__
(self)
draggan/stylegan2/training/dataset.py:75
↓ 1 callersMethod__init__
(self, channel, bias=True, negative_slope=0.2, scale=2 ** 0.5)
draggan/deprecated/stylegan2/op/fused_act.py:103
↓ 1 callersMethod__init__
(self, name, # Name of the dataset. raw_shape, # Shape of the r
draggan/stylegan2/training/dataset.py:27
↓ 1 callersMethod__reduce__
(self)
draggan/stylegan2/torch_utils/persistence.py:118
↓ 1 callersFunction_bias_act_cuda
Fast CUDA implementation of `bias_act()` using custom ops.
draggan/stylegan2/torch_utils/ops/bias_act.py:129
↓ 1 callersFunction_bias_act_ref
Slow reference implementation of `bias_act()` using standard TensorFlow ops.
draggan/stylegan2/torch_utils/ops/bias_act.py:94
↓ 1 callersFunction_check_pickleable
r"""Check that the given object is pickleable, raising an exception if it is not. This function is expected to be considerably more efficient
draggan/stylegan2/torch_utils/persistence.py:231
↓ 1 callersFunction_find_compiler_bindir
()
draggan/stylegan2/torch_utils/custom_ops.py:28
↓ 1 callersFunction_init
()
draggan/stylegan2/torch_utils/ops/bias_act.py:41
↓ 1 callersFunction_init
()
draggan/stylegan2/torch_utils/ops/upfirdn2d.py:26
↓ 1 callersMethod_load_raw_image
(self, raw_idx)
draggan/stylegan2/training/dataset.py:69
↓ 1 callersMethod_load_raw_image
(self, raw_idx)
draggan/stylegan2/training/dataset.py:212
↓ 1 callersMethod_load_raw_labels
(self)
draggan/stylegan2/training/dataset.py:72
↓ 1 callersFunction_module_to_src
r"""Query the source code of a given Python module.
draggan/stylegan2/torch_utils/persistence.py:206
↓ 1 callersFunction_should_use_custom_op
()
draggan/stylegan2/torch_utils/ops/grid_sample_gradfix.py:34
↓ 1 callersFunction_src_to_module
r"""Get or create a Python module for the given source code.
draggan/stylegan2/torch_utils/persistence.py:216
next →1–100 of 447, ranked by callers