MCPcopy Create free account

hub / github.com/NVlabs/stylegan3 / functions

Functions485 in github.com/NVlabs/stylegan3

↓ 47 callersFunctionkwarg
(tf_name, default=None, none=None)
legacy.py:114
↓ 34 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
↓ 29 callersMethodappend
(self, x)
metrics/metric_utils.py:94
↓ 17 callersMethodupdate
(self, cur_items)
metrics/metric_utils.py:167
↓ 15 callersMethodload
(self, pkl, ignore_errors=False)
viz/pickle_widget.py:47
↓ 11 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
training/networks_stylegan2.py:528
↓ 9 callersFunctionsinc
(x)
metrics/equivariance.py:22
↓ 8 callersFunctionerror
(msg)
dataset_tool.py:31
↓ 8 callersMethodsub
(self, tag=None, num_items=None, flush_interval=1000, rel_lo=0, rel_hi=1)
metrics/metric_utils.py:182
↓ 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_parse_padding
(padding)
torch_utils/ops/upfirdn2d.py:44
↓ 6 callersFunction_parse_scaling
(scaling)
torch_utils/ops/upfirdn2d.py:35
↓ 6 callersMethodsave
(self, pkl_file)
metrics/metric_utils.py:137
↓ 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
↓ 6 callersMethodupdate
(self, image)
gui_utils/gl_utils.py:179
↓ 5 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
training/networks_stylegan3.py:491
↓ 5 callersFunction_get_filter_size
(f)
torch_utils/ops/upfirdn2d.py:55
↓ 5 callersFunction_sinc
(x)
viz/renderer.py:44
↓ 5 callersFunctioniterate_images
()
dataset_tool.py:85
↓ 5 callersFunctionmaybe_min
(a: int, b: Optional[int])
dataset_tool.py:51
↓ 5 callersFunctionscale2d_inv
(sx, sy, **kwargs)
training/augment.py:108
↓ 5 callersMethodskip_frame
(self)
gui_utils/glfw_window.py:130
↓ 4 callersFunction_conv2d_gradfix
(transpose, weight_shape, stride, padding, output_padding, dilation, groups)
torch_utils/ops/conv2d_gradfix.py:71
↓ 4 callersMethod_get_raw_labels
(self)
training/dataset.py:53
↓ 4 callersFunction_lanczos_window
(x, a)
viz/renderer.py:49
↓ 4 callersFunction_tuple_of_ints
(xs, ndim)
torch_utils/ops/conv2d_gradfix.py:60
↓ 4 callersMethodbackward
(ctx, dy)
torch_utils/ops/bias_act.py:158
↓ 4 callersMethodend_frame
(self)
gui_utils/glfw_window.py:198
↓ 4 callersMethodflush
Flush written text to both stdout and a file, if open.
dnnlib/util.py:93
↓ 4 callersMethodget_all
(self)
metrics/metric_utils.py:123
↓ 4 callersFunctionget_texture_format
(dtype, channels)
gui_utils/gl_utils.py:76
↓ 4 callersFunctionlanczos_window
(x, a)
metrics/equivariance.py:27
↓ 4 callersMethodrender
(self, **args)
viz/renderer.py:131
↓ 4 callersMethodstd
r"""Returns the standard deviation of the scalars that were accumulated for the given statistic between the last two calls to `update(
torch_utils/training_stats.py:198
↓ 4 callersMethodwrite
Write text to stdout (and a file) and optionally flush.
dnnlib/util.py:78
↓ 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 callersFunction_unbroadcast
(x, shape)
torch_utils/ops/fma.py:49
↓ 3 callersMethodbind
(self)
gui_utils/gl_utils.py:173
↓ 3 callersFunctionfile_ext
(name: Union[str, Path])
dataset_tool.py:58
↓ 3 callersMethodget_label
(self, idx)
training/dataset.py:97
↓ 3 callersFunctionget_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
dnnlib/util.py:277
↓ 3 callersMethodis_compatible
(self, *, image=None, width=None, height=None, channels=None, dtype=None)
gui_utils/gl_utils.py:201
↓ 3 callersMethodmake_context_current
(self)
gui_utils/glfw_window.py:157
↓ 3 callersFunctionnamed_params_and_buffers
(module)
torch_utils/misc.py:151
↓ 3 callersFunctionprepare_texture_data
(image)
gui_utils/gl_utils.py:81
↓ 3 callersFunctionrecurse
(prefix, tf_net)
legacy.py:76
↓ 3 callersFunctionrotate2d_inv
(theta, **kwargs)
training/augment.py:111
↓ 3 callersFunctionrotation_matrix
(angle)
metrics/equivariance.py:31
↓ 3 callersMethodrun_D
(self, img, c, blur_sigma=0, update_emas=False)
training/loss.py:52
↓ 3 callersMethodrun_G
(self, z, c, update_emas=False)
training/loss.py:42
↓ 3 callersFunctionsave_image_grid
(img, fname, drange, grid_size)
training/training_loop.py:70
↓ 3 callersFunctionscale2d
(sx, sy, **kwargs)
training/augment.py:73
↓ 3 callersMethodto_device
(self, buf)
viz/renderer.py:203
↓ 3 callersFunctiontranslate2d
(tx, ty, **kwargs)
training/augment.py:58
↓ 3 callersFunctiontranslate2d_inv
(tx, ty, **kwargs)
training/augment.py:105
↓ 3 callersFunctionupfirdn2d
r"""Pad, upsample, filter, and downsample a batch of 2D images. Performs the following sequence of operations for each channel: 1. Upsample
torch_utils/ops/upfirdn2d.py:118
↓ 2 callersMethod__init__
(self)
viz/renderer.py:120
↓ 2 callersMethod_adjust_font_size
(self)
visualizer.py:100
↓ 2 callersMethod_attach_glfw_callbacks
(self)
gui_utils/glfw_window.py:218
↓ 2 callersFunction_collect_tf_params
(tf_net)
legacy.py:73
↓ 2 callersFunction_filtered_lrelu_cuda
Fast CUDA implementation of `filtered_lrelu()` using custom ops.
torch_utils/ops/filtered_lrelu.py:159
↓ 2 callersFunction_get_array_priv
( string: str, *, size: int = 32, max_width: Optional[int]=None, max_height: Optional[int]=Non
gui_utils/text_utils.py:45
↓ 2 callersFunction_get_filter_size
(f)
torch_utils/ops/filtered_lrelu.py:35
↓ 2 callersMethod_get_pinned_buf
(self, ref)
viz/renderer.py:195
↓ 2 callersFunction_get_weight_shape
(w)
torch_utils/ops/conv2d_resample.py:21
↓ 2 callersMethod_get_zipfile
(self)
training/dataset.py:189
↓ 2 callersMethod_ignore_timing
(self)
viz/renderer.py:209
↓ 2 callersMethod_open_file
(self, fname)
training/dataset.py:195
↓ 2 callersFunction_parse_padding
(padding)
torch_utils/ops/filtered_lrelu.py:42
↓ 2 callersFunction_populate_module_params
(module, *patterns)
legacy.py:86
↓ 2 callersFunction_should_use_custom_op
(input)
torch_utils/ops/conv2d_gradfix.py:49
↓ 2 callersFunction_upfirdn2d_cuda
Fast CUDA implementation of `upfirdn2d()` using custom ops.
torch_utils/ops/upfirdn2d.py:217
↓ 2 callersMethodadd_recent_pickle
(self, pkl, ignore_errors=False)
visualizer.py:74
↓ 2 callersMethodappend_torch
(self, x, num_gpus=1, rank=0)
metrics/metric_utils.py:111
↓ 2 callersMethodbegin_frame
(self)
gui_utils/glfw_window.py:161
↓ 2 callersMethodbind
(self)
gui_utils/gl_utils.py:290
↓ 2 callersFunctionbutton
(label, width=0, enabled=True)
gui_utils/imgui_utils.py:90
↓ 2 callersFunctioncalc_output_padding
(input_shape, output_shape)
torch_utils/ops/conv2d_gradfix.py:98
↓ 2 callersMethodclear_result
(self)
visualizer.py:89
↓ 2 callersMethodclose
(self)
training/dataset.py:66
↓ 2 callersMethodclose
Flush, close possible files, and remove stdout/stderr mirroring.
dnnlib/util.py:100
↓ 2 callersFunctioncompute_distances
(row_features, col_features, num_gpus, rank, col_batch_size)
metrics/precision_recall.py:19
↓ 2 callersFunctionconstruct_affine_bandlimit_filter
(mat, a=3, amax=16, aflt=64, up=4, cutoff_in=1, cutoff_out=1)
metrics/equivariance.py:102
↓ 2 callersFunctionconstruct_heatmap
(npz_file, smooth)
avg_spectra.py:65
↓ 2 callersFunctionconvert_tf_generator
(tf_G)
legacy.py:107
↓ 2 callersMethoddefer_rendering
(self, num_frames=1)
visualizer.py:86
↓ 2 callersMethoddesign_lowpass_filter
(numtaps, cutoff, width, fs, radial=False)
training/networks_stylegan3.py:364
↓ 2 callersMethoddrag
(self, dx, dy)
viz/latent_widget.py:23
↓ 2 callersFunctiondrag_previous_control
(enabled=True)
gui_utils/imgui_utils.py:137
↓ 2 callersMethoddraw
(self, *, pos=0, zoom=1, align=0, rint=False, color=1, alpha=1, rounding=0)
gui_utils/gl_utils.py:192
↓ 2 callersFunctiondraw_shape
(vertices, *, mode=gl.GL_TRIANGLE_FAN, pos=0, size=1, color=1, alpha=1)
gui_utils/gl_utils.py:308
↓ 2 callersMethoddump_png
(self, image)
viz/capture_widget.py:28
↓ 2 callersFunctiongenerate_image
(seed)
avg_spectra.py:39
↓ 2 callersMethodget_all_torch
(self)
metrics/metric_utils.py:127
↓ 2 callersFunctionget_feature_detector
(url, device=torch.device('cpu'), num_gpus=1, rank=0, verbose=False)
metrics/metric_utils.py:42
↓ 2 callersMethodget_mean_cov
(self)
metrics/metric_utils.py:130
↓ 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
dnnlib/util.py:236
↓ 2 callersFunctionget_pil_font
(font=None, size=32)
gui_utils/text_utils.py:29
next →1–100 of 485, ranked by callers