MCPcopy Create free account

hub / github.com/XingangPan/DragGAN / functions

Functions1,199 in github.com/XingangPan/DragGAN

↓ 69 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
stylegan_human/torch_utils/training_stats.py:190
↓ 51 callersMethodupdate
(self, image)
gui_utils/gl_utils.py:180
↓ 47 callersFunctionkwarg
(tf_name, default=None, none=None)
legacy.py:114
↓ 42 callersMethodload
(self, pkl, ignore_errors=False)
viz/pickle_widget.py:47
↓ 20 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
torch_utils/training_stats.py:188
↓ 20 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
stylegan_human/torch_utils/training_stats.py:149
↓ 16 callersMethod__init__
(self, kernel, pad, upsample_factor=1)
stylegan_human/torch_utils/models_face.py:79
↓ 16 callersMethod__init__
(self, kernel, pad, upsample_factor=1)
stylegan_human/torch_utils/models.py:76
↓ 16 callersFunction_shape
(tf_expr, dim_idx)
stylegan_human/dnnlib/tflib/ops/upfirdn_2d.py:339
↓ 15 callersMethod__init__
(self, kernel, pad, upsample_factor=1)
stylegan_human/pti/pti_models/e4e/stylegan2/model.py:73
↓ 14 callersMethodclone
Create a clone of this network with its own copy of the variables.
stylegan_human/dnnlib/tflib/network.py:303
↓ 13 callersMethodrun
Run this network for the given NumPy array(s), and return the output(s) as NumPy array(s). Args: input_transform: A dict speci
stylegan_human/dnnlib/tflib/network.py:355
↓ 12 callersMethodbackward
(ctx, dy)
stylegan_human/torch_utils/ops/bias_act.py:163
↓ 12 callersFunctionget_block
(in_channel, depth, num_units, stride=2)
stylegan_human/pti/pti_models/e4e/encoders/helpers.py:26
↓ 11 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
stylegan_human/training/networks_stylegan2.py:534
↓ 11 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
stylegan_human/training_scripts/sg2/training/networks.py:509
↓ 11 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
stylegan_human/training_scripts/sg3/training/networks_stylegan2.py:554
↓ 11 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
training/networks_stylegan2.py:534
↓ 11 callersMethodconvert
Create new network with the given parameters, and copy all variables from this network.
stylegan_human/dnnlib/tflib/network.py:333
↓ 10 callersFunctionupdate_image_draw
(image, points, mask, show_mask, global_state=None)
visualizer_drag_gradio.py:102
↓ 9 callersFunctionupfirdn2d
r"""Pad, upsample, filter, and downsample a batch of 2D images. Performs the following sequence of operations for each channel: 1. Upsample
stylegan_human/torch_utils/ops/upfirdn2d.py:122
↓ 7 callersFunctionmatrix
(*rows, device=None)
stylegan_human/training/augment.py:48
↓ 7 callersFunctionmatrix
(*rows, device=None)
training/augment.py:48
↓ 6 callersFunction_conv2d_wrapper
Wrapper for the underlying `conv2d()` and `conv_transpose2d()` implementations.
torch_utils/ops/conv2d_resample.py:29
↓ 6 callersFunction_conv2d_wrapper
Wrapper for the underlying `conv2d()` and `conv_transpose2d()` implementations.
stylegan_human/torch_utils/ops/conv2d_resample.py:31
↓ 6 callersFunction_parse_padding
(padding)
torch_utils/ops/upfirdn2d.py:44
↓ 6 callersFunction_parse_padding
(padding)
stylegan_human/torch_utils/ops/upfirdn2d.py:48
↓ 6 callersFunction_parse_scaling
(scaling)
torch_utils/ops/upfirdn2d.py:35
↓ 6 callersFunction_parse_scaling
(scaling)
stylegan_human/torch_utils/ops/upfirdn2d.py:39
↓ 6 callersFunctionget_image_from_w
(w, G)
stylegan_human/utils/log_utils.py:76
↓ 6 callersMethodtrain
(self)
stylegan_human/pti/training/coaches/base_coach.py:98
↓ 6 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
torch_utils/training_stats.py:147
↓ 5 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
stylegan_human/training/networks_stylegan3.py:492
↓ 5 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
stylegan_human/training_scripts/sg3/training/networks_stylegan3.py:513
↓ 5 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
training/networks_stylegan3.py:497
↓ 5 callersFunction_get_filter_size
(f)
torch_utils/ops/upfirdn2d.py:55
↓ 5 callersFunction_get_filter_size
(f)
stylegan_human/torch_utils/ops/upfirdn2d.py:59
↓ 5 callersFunction_setup_kernel
(k)
stylegan_human/dnnlib/tflib/ops/upfirdn_2d.py:346
↓ 5 callersFunction_simple_upfirdn_2d
(x, k, up=1, down=1, pad0=0, pad1=0, data_format='NCHW', impl='cuda')
stylegan_human/dnnlib/tflib/ops/upfirdn_2d.py:355
↓ 5 callersMethodbackward
(ctx, dy)
torch_utils/ops/bias_act.py:158
↓ 5 callersFunctionclear_state
Clear target history state from global_state If target is not defined, points and mask will be both removed. 1. set global_state['points'] as
visualizer_drag_gradio.py:39
↓ 5 callersFunctioninit_images
This function is called only ones with Gradio App is started. 0. pre-process global_state, unpack value from global_state of need 1. Re-init r
visualizer_drag_gradio.py:61
↓ 5 callersFunctionscale2d_inv
(sx, sy, **kwargs)
stylegan_human/training/augment.py:108
↓ 5 callersFunctionscale2d_inv
(sx, sy, **kwargs)
training/augment.py:108
↓ 4 callersFunction_conv2d_gradfix
(transpose, weight_shape, stride, padding, output_padding, dilation, groups)
torch_utils/ops/conv2d_gradfix.py:66
↓ 4 callersFunction_conv2d_gradfix
(transpose, weight_shape, stride, padding, output_padding, dilation, groups)
stylegan_human/torch_utils/ops/conv2d_gradfix.py:70
↓ 4 callersMethod_get_raw_labels
(self)
stylegan_human/training/dataset.py:53
↓ 4 callersMethod_get_raw_labels
(self)
stylegan_human/training_scripts/sg2/training/dataset.py:59
↓ 4 callersMethod_get_raw_labels
(self)
stylegan_human/training_scripts/sg3/training/dataset.py:62
↓ 4 callersMethod_get_raw_labels
(self)
training/dataset.py:53
↓ 4 callersFunction_tuple_of_ints
(xs, ndim)
torch_utils/ops/conv2d_gradfix.py:55
↓ 4 callersFunction_tuple_of_ints
(xs, ndim)
stylegan_human/torch_utils/ops/conv2d_gradfix.py:60
↓ 4 callersFunction_upsample_add
Upsample and add two feature maps. Args: x: (Variable) top feature map to be upsampled. y: (Variable) lateral feature map. Returns
stylegan_human/pti/pti_models/e4e/encoders/helpers.py:123
↓ 4 callersFunctionassert_tf_initialized
Check that TensorFlow session has been initialized.
stylegan_human/dnnlib/tflib/tfutil.py:137
↓ 4 callersMethodclose
Flush, close possible files, and remove stdout/stderr mirroring.
stylegan_human/dnnlib/util.py:101
↓ 4 callersFunctiondecoder
(G, style_space, latent, noise)
stylegan_human/edit/edit_helper.py:58
↓ 4 callersMethoddetect_face_dlib
(self, img)
stylegan_human/insetgan.py:104
↓ 4 callersMethodend_frame
(self)
gui_utils/glfw_window.py:198
↓ 4 callersMethodflush
Flush written text to both stdout and a file, if open.
stylegan_human/dnnlib/util.py:94
↓ 4 callersMethodflush
Flush written text to both stdout and a file, if open.
dnnlib/util.py:93
↓ 4 callersFunctionget_texture_format
(dtype, channels)
gui_utils/gl_utils.py:77
↓ 4 callersMethodis_compatible
(self, *, image=None, width=None, height=None, channels=None, dtype=None)
gui_utils/gl_utils.py:202
↓ 4 callersMethodloss_border
(self, A_face, B, p1=10000, p2=2, spec=None)
stylegan_human/insetgan.py:81
↓ 4 callersMethodloss_coarse
(self, A_face, B, p1=500, p2=0.05)
stylegan_human/insetgan.py:59
↓ 4 callersMethodmean_latent
(self, n_latent)
stylegan_human/torch_utils/models.py:455
↓ 4 callersFunctionpreprocess_mask_info
Function to handle mask information. 1. last_mask is None: Do not need to change mask, return mask 2. last_mask is not None: 2.1 globa
visualizer_drag_gradio.py:115
↓ 4 callersMethodrender
(self, **args)
viz/renderer.py:85
↓ 4 callersMethodskip_frame
(self)
gui_utils/glfw_window.py:130
↓ 4 callersMethodwrite
Write text to stdout (and a file) and optionally flush.
stylegan_human/dnnlib/util.py:79
↓ 3 callersMethod_file_ext
(fname)
stylegan_human/training/dataset.py:186
↓ 3 callersMethod_file_ext
(fname)
stylegan_human/training_scripts/sg2/training/dataset.py:204
↓ 3 callersMethod_file_ext
(fname)
stylegan_human/training_scripts/sg3/training/dataset.py:208
↓ 3 callersMethod_file_ext
(fname)
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
torch_utils/training_stats.py:170
↓ 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
stylegan_human/torch_utils/training_stats.py:172
↓ 3 callersMethod_init_fields
(self)
stylegan_human/dnnlib/tflib/network.py:102
↓ 3 callersMethod_init_graph
(self)
stylegan_human/dnnlib/tflib/network.py:127
↓ 3 callersMethod_render_drag_impl
(self, res, points = [], targets = [], mask = None,
viz/renderer.py:272
↓ 3 callersFunction_unbroadcast
(x, shape)
torch_utils/ops/fma.py:49
↓ 3 callersFunction_unbroadcast
(x, shape)
stylegan_human/torch_utils/ops/fma.py:51
↓ 3 callersFunctionadd_watermark_np
(input_image_array, watermark_text="AI Generated")
viz/renderer.py:48
↓ 3 callersMethodbind
(self)
gui_utils/gl_utils.py:174
↓ 3 callersFunctionconv_warper
(layer, input, style, noise)
stylegan_human/edit/edit_helper.py:10
↓ 3 callersMethoddelete
(self)
gui_utils/gl_utils.py:162
↓ 3 callersMethoddraw
(self, *, pos=0, zoom=1, align=0, rint=False, color=1, alpha=1, rounding=0)
gui_utils/gl_utils.py:193
↓ 3 callersMethodfind_var
Find variable by local or global name.
stylegan_human/dnnlib/tflib/network.py:242
↓ 3 callersFunctionfused_leaky_relu
(input, bias, negative_slope=0.2, scale=2 ** 0.5)
stylegan_human/torch_utils/op_edit/fused_act.py:88
↓ 3 callersFunctiongenerate_image_from_z
(G, z, noise_mode, truncation_psi, device)
stylegan_human/interpolation.py:49
↓ 3 callersFunctionget_blocks
(num_layers)
stylegan_human/pti/pti_models/e4e/encoders/helpers.py:30
↓ 3 callersFunctionget_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
stylegan_human/dnnlib/util.py:264
↓ 3 callersFunctionget_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
dnnlib/util.py:277
↓ 3 callersFunctionlog_image_from_w
(w, G, name)
stylegan_human/utils/log_utils.py:12
↓ 3 callersMethodmake_context_current
(self)
gui_utils/glfw_window.py:157
↓ 3 callersFunctionmake_kernel
(k)
stylegan_human/torch_utils/models_face.py:25
↓ 3 callersFunctionmake_kernel
(k)
stylegan_human/torch_utils/models.py:27
↓ 3 callersFunctionmake_kernel
(k)
stylegan_human/pti/pti_models/e4e/stylegan2/model.py:19
↓ 3 callersFunctionmake_layers
(block, no_relu_layers)
stylegan_human/openpose/src/model.py:7
↓ 3 callersFunctionnamed_params_and_buffers
(module)
torch_utils/misc.py:151
↓ 3 callersFunctionnamed_params_and_buffers
(module)
stylegan_human/torch_utils/misc.py:151
↓ 3 callersFunctionprepare_texture_data
(image)
gui_utils/gl_utils.py:82
next →1–100 of 1,199, ranked by callers