MCPcopy Create free account

hub / github.com/autonomousvision/stylegan-xl / functions

Functions736 in github.com/autonomousvision/stylegan-xl

⨍ Functions 736◇ Types & classes 120↳ Endpoints 8
↓ 2 callersFunction_populate_module_params
(module, *patterns)
legacy.py:92
↓ 2 callersFunction_should_use_custom_op
(input)
torch_utils/ops/conv2d_gradfix.py:47
↓ 2 callersFunction_transform
(n_px)
feature_networks/clip/clip.py:59
↓ 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:125
↓ 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 callersFunctionbytes_to_unicode
Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you
feature_networks/clip/simple_tokenizer.py:16
↓ 2 callersFunctioncalc_output_padding
(input_shape, output_shape)
torch_utils/ops/conv2d_gradfix.py:93
↓ 2 callersMethodclear_result
(self)
visualizer.py:89
↓ 2 callersMethodclose
(self)
training/dataset.py:80
↓ 2 callersFunctioncompute_distances
(row_features, col_features, num_gpus, rank, col_batch_size)
metrics/precision_recall.py:20
↓ 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 callersFunctionconvert_tf_generator
(tf_G)
legacy.py:113
↓ 2 callersMethoddefer_rendering
(self, num_frames=1)
visualizer.py:86
↓ 2 callersMethoddesign_lowpass_filter
(numtaps, cutoff, width, fs, radial=False)
training/networks_stylegan3.py:376
↓ 2 callersMethoddesign_lowpass_filter
(numtaps, cutoff, width, fs, radial=False)
training/networks_stylegan3_resetting.py:427
↓ 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 callersMethodencode_conv_features
(self, features)
feature_networks/clip/model.py:354
↓ 2 callersFunctionforward_vit
(pretrained, x)
feature_networks/vit.py:56
↓ 2 callersFunctiongetActivation
(name)
metrics/metric_utils.py:211
↓ 2 callersMethodget_all_torch
(self)
metrics/metric_utils.py:141
↓ 2 callersFunctionget_backbone_normstats
(backbone)
pg_modules/projector.py:9
↓ 2 callersMethodget_mean_cov
(self)
metrics/metric_utils.py:144
↓ 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_pairs
Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).
feature_networks/clip/simple_tokenizer.py:38
↓ 2 callersFunctionget_pil_font
(font=None, size=32)
gui_utils/text_utils.py:29
↓ 2 callersFunctionget_readout_oper
(vit_features, features, use_readout, start_index=1)
feature_networks/vit.py:166
↓ 2 callersFunctiongrayed_out
(cond=True)
gui_utils/imgui_utils.py:42
↓ 2 callersFunctioninterpolation_checks
Tests for the interpolation functions
torch_utils/gen_utils.py:158
↓ 2 callersFunctionis_image_ext
(fname: Union[str, Path])
dataset_tool.py:64
↓ 2 callersFunctionis_persistent
r"""Test whether the given object or class is persistent, i.e., whether it will save its source code when pickled.
torch_utils/persistence.py:134
↓ 2 callersFunctionis_valid_metric
(metric)
metrics/metric_main.py:29
↓ 2 callersMethodload
(pkl_file)
metrics/metric_utils.py:156
↓ 2 callersFunctionmodulated_conv2d
( x, # Input tensor of shape [batch_size, in_channels, in_height, in_width].
training/networks_stylegan2.py:32
↓ 2 callersMethodnames
r"""Returns the names of all statistics broadcasted so far that match the regular expression specified at construction time.
torch_utils/training_stats.py:141
↓ 2 callersFunctionnormalize_2nd_moment
(x, dim=1, eps=1e-8)
training/networks_stylegan2.py:26
↓ 2 callersFunctionnormalize_second_moment
(x, dim=1, eps=1e-8)
training/networks_fastgan.py:9
↓ 2 callersFunctionpatch_device
(module)
feature_networks/clip/clip.py:124
↓ 2 callersFunctionpatch_float
(module)
feature_networks/clip/clip.py:144
↓ 2 callersMethodreinit_stem
(self)
training/networks_stylegan3_resetting.py:673
↓ 2 callersMethodresolve_pkl
(self, pattern)
viz/pickle_widget.py:148
↓ 2 callersMethodrun_synthesis_net
(net, *args, capture_layer=None, **kwargs)
viz/renderer.py:341
↓ 2 callersFunctionsave_image
(img, path)
run_stylemc.py:67
↓ 2 callersMethodset_position
(self, x, y)
gui_utils/glfw_window.py:112
↓ 2 callersMethodset_vsync
(self, vsync)
gui_utils/glfw_window.py:118
↓ 2 callersMethodset_window_size
(self, width, height)
gui_utils/glfw_window.py:99
↓ 2 callersFunctionslerp
Spherical linear interpolation between v0 (starting) and v1 (final) vectors; for optimal results, use t as an np.ndarray to return all result
torch_utils/gen_utils.py:188
↓ 2 callersFunctionslerp
(a, b, t)
metrics/perceptual_path_length.py:22
↓ 2 callersFunctionsuppress_tracer_warnings
()
torch_utils/misc.py:71
↓ 2 callersMethodto_cpu
(self, buf)
viz/renderer.py:206
↓ 1 callersFunctionDiffAugment
(x, policy='', channels_first=True)
training/diffaug.py:9
↓ 1 callersMethod__getstate__
(self)
training/dataset.py:89
↓ 1 callersMethod__init__
(self, name, # Name of the dataset. raw_shape, # Shape of the r
training/dataset.py:28
↓ 1 callersMethod__init__
(self, *, title='ImguiWindow', font=None, font_sizes=range(14,24), **glfw_kwargs)
gui_utils/imgui_window.py:20
↓ 1 callersMethod__reduce__
(self)
torch_utils/persistence.py:118
↓ 1 callersFunction_apply_affine_transformation
(x, mat, up=4, **filter_kwargs)
viz/renderer.py:87
↓ 1 callersMethod_apply_cmap
(self, x, name='viridis')
viz/renderer.py:212
↓ 1 callersFunction_bias_act_cuda
Fast CUDA implementation of `bias_act()` using custom ops.
torch_utils/ops/bias_act.py:126
↓ 1 callersFunction_bias_act_ref
Slow reference implementation of `bias_act()` using standard TensorFlow ops.
torch_utils/ops/bias_act.py:91
↓ 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
torch_utils/persistence.py:231
↓ 1 callersFunction_construct_affine_bandlimit_filter
(mat, a=3, amax=16, aflt=64, up=4, cutoff_in=1, cutoff_out=1)
viz/renderer.py:55
↓ 1 callersFunction_download
(url: str, root: str = os.path.expanduser("~/.cache/clip"))
feature_networks/clip/clip.py:27
↓ 1 callersFunction_filtered_lrelu_ref
Slow and memory-inefficient reference implementation of `filtered_lrelu()` using existing `upfirdn2n()` and `bias_act()` ops.
torch_utils/ops/filtered_lrelu.py:121
↓ 1 callersFunction_find_compiler_bindir
()
torch_utils/custom_ops.py:29
↓ 1 callersFunction_get_array_impl
(string, *, font=None, size=32, outline=0, outline_pad=3, outline_coef=3, outline_exp=2, line_pad: int=None)
gui_utils/text_utils.py:74
↓ 1 callersFunction_get_array_impl_dropshadow
(string, *, font=None, size=32, radius: int, offset_x: int, offset_y: int, line_pad: int=None, **kwargs)
gui_utils/text_utils.py:96
↓ 1 callersMethod_get_batch_fmtstr
(self, num_batches)
run_stylemc.py:61
↓ 1 callersFunction_get_mangled_gpu_name
()
torch_utils/custom_ops.py:44
↓ 1 callersFunction_init
()
torch_utils/ops/filtered_lrelu.py:23
↓ 1 callersFunction_init
()
torch_utils/ops/bias_act.py:38
↓ 1 callersFunction_init
()
torch_utils/ops/upfirdn2d.py:23
↓ 1 callersMethod_load_raw_image
(self, raw_idx)
training/dataset.py:83
↓ 1 callersMethod_load_raw_image
(self, raw_idx)
training/dataset.py:226
↓ 1 callersFunction_locate_results
(pattern)
viz/pickle_widget.py:22
↓ 1 callersFunction_make_ghostnet
(model)
feature_networks/pretrained_builder.py:128
↓ 1 callersFunction_make_pretrained
(backbone, verbose=False)
feature_networks/pretrained_builder.py:195
↓ 1 callersFunction_make_projector
(im_res, backbone, cout, proj_type, expand=False)
pg_modules/projector.py:55
↓ 1 callersFunction_make_regnet
(model)
feature_networks/pretrained_builder.py:32
↓ 1 callersFunction_make_resnet_clip
(model)
feature_networks/pretrained_builder.py:60
↓ 1 callersFunction_make_scratch_ccm
(scratch, in_channels, cout, expand=False)
pg_modules/projector.py:31
↓ 1 callersFunction_make_scratch_csm
(scratch, in_channels, cout, expand)
pg_modules/projector.py:44
↓ 1 callersFunction_make_shufflenet
(model)
feature_networks/pretrained_builder.py:102
↓ 1 callersFunction_make_vit
(model, name)
feature_networks/pretrained_builder.py:138
↓ 1 callersFunction_make_vit_b_rn50_backbone
( model, features=[256, 512, 768, 768], size=[384, 384], hooks=[0, 1, 8, 11], vit_features
feature_networks/vit.py:343
↓ 1 callersFunction_module_to_src
r"""Query the source code of a given Python module.
torch_utils/persistence.py:206
↓ 1 callersMethod_render_impl
(self, res, pkl = None, w0_seeds = [[0, 1]], stylemix_idx = [],
viz/renderer.py:224
↓ 1 callersMethod_set_args_async
(self, **args)
visualizer.py:215
↓ 1 callersMethod_set_args_sync
(self, **args)
visualizer.py:227
↓ 1 callersFunction_setup_circle
(hole)
gui_utils/gl_utils.py:368
↓ 1 callersFunction_setup_rect
(rx, ry)
gui_utils/gl_utils.py:352
↓ 1 callersFunction_should_use_custom_op
()
torch_utils/ops/grid_sample_gradfix.py:33
↓ 1 callersFunction_src_to_module
r"""Get or create a Python module for the given source code.
torch_utils/persistence.py:216
↓ 1 callersFunction_sync
r"""Synchronize the global cumulative counters across devices and processes. Called internally by `Collector.update()`.
torch_utils/training_stats.py:234
← previousnext →101–200 of 736, ranked by callers