MCPcopy Create free account

hub / github.com/NVlabs/stylegan / functions

Functions280 in github.com/NVlabs/stylegan

↓ 39 callersMethodget_output_for
Construct TensorFlow expression(s) for the output(s) of this network, given the input expression(s).
dnnlib/tflib/network.py:200
↓ 31 callersFunctionautosummary
Create a new autosummary. Args: name: Name to use in TensorBoard value: TensorFlow expression or python value to track
dnnlib/tflib/autosummary.py:74
↓ 23 callersMethodrun
(self)
dataset_tool.py:118
↓ 21 callersFunctionfp32
(*values)
training/loss.py:17
↓ 13 callersFunctionadd_command
(cmd, desc, example=None)
dataset_tool.py:555
↓ 13 callersFunctionapply_bias
(x)
training/networks_progan.py:57
↓ 13 callersFunctionnf
(stage)
training/networks_progan.py:172
↓ 13 callersMethodupdate
Do general housekeeping and keep the state of the context up-to-date. Should be called often enough but not in a tight loop.
dnnlib/submission/run_context.py:61
↓ 11 callersFunctionnf
(stage)
training/networks_stylegan.py:466
↓ 10 callersMethodadd_image
(self, img)
dataset_tool.py:66
↓ 10 callersFunctionapply_bias
(x, lrmul=1)
training/networks_stylegan.py:213
↓ 10 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
dnnlib/tflib/network.py:354
↓ 9 callersMethodclone
Create a clone of this network with its own copy of the variables.
dnnlib/tflib/network.py:302
↓ 8 callersFunctionblock
(res, x)
training/networks_stylegan.py:517
↓ 8 callersFunctionconv2d
(x, fmaps, kernel, gain=np.sqrt(2), use_wscale=False)
training/networks_progan.py:48
↓ 8 callersFunctionconv2d
(x, fmaps, kernel, **kwargs)
training/networks_stylegan.py:164
↓ 8 callersFunctionload_Gs
(url)
generate_figures.py:31
↓ 7 callersMethodchoose_shuffled_order
(self)
dataset_tool.py:61
↓ 7 callersMethodget_minibatch_np
(self, minibatch_size, lod=0)
training/dataset.py:149
↓ 6 callersFunctionblock
(x, res)
training/networks_progan.py:185
↓ 6 callersMethodclose
Close the context and clean up. Should only be called once.
dnnlib/submission/run_context.py:90
↓ 6 callersMethodfind_var
Find variable by local or global name.
dnnlib/tflib/network.py:241
↓ 6 callersFunctionfunc
(x)
training/networks_stylegan.py:99
↓ 6 callersMethodrun
(self, network_pkl, run_dir=None, dataset_args=None, mirror_augment=None, num_gpus=1, tf_config=None, log_resu
metrics/metric_base.py:45
↓ 6 callersFunctiontorgb
(res, x)
training/networks_stylegan.py:524
↓ 5 callersFunctionPN
(x)
training/networks_progan.py:173
↓ 5 callersMethod_report_result
(self, value, suffix='', fmt='%-10.4f')
metrics/metric_base.py:90
↓ 5 callersMethodadd_labels
(self, labels)
dataset_tool.py:91
↓ 5 callersMethodapply_loss_scaling
Apply dynamic loss scaling for the given expression.
dnnlib/tflib/optimizer.py:198
↓ 5 callersFunctiondense
(x, fmaps, **kwargs)
training/networks_stylegan.py:154
↓ 5 callersFunctionerror
(msg)
dataset_tool.py:27
↓ 5 callersFunctionfromrgb
(x, res)
training/networks_stylegan.py:599
↓ 5 callersFunctionlayer_epilogue
(x, layer_idx)
training/networks_stylegan.py:490
↓ 5 callersFunctiontorgb
(x, res)
training/networks_progan.py:206
↓ 5 callersMethodundo_loss_scaling
Undo the effect of dynamic loss scaling for the given expression.
dnnlib/tflib/optimizer.py:207
↓ 5 callersFunctionupscale2d
(x, factor=2)
training/networks_progan.py:75
↓ 5 callersFunctionupscale2d
(x, factor=2)
training/networks_stylegan.py:108
↓ 4 callersFunction_blur2d
(x, f=[1,2,1], normalize=True, flip=False, stride=1)
training/networks_stylegan.py:22
↓ 4 callersFunctionassert_tf_initialized
Check that TensorFlow session has been initialized.
dnnlib/tflib/tfutil.py:122
↓ 4 callersFunctioncset
(cur_lambda, new_cond, new_lambda)
training/networks_progan.py:20
↓ 4 callersFunctioncset
(cur_lambda, new_cond, new_lambda)
training/networks_stylegan.py:548
↓ 4 callersFunctiondownscale2d
(x, factor=2)
training/networks_progan.py:102
↓ 4 callersFunctiondownscale2d
(x, factor=2)
training/networks_stylegan.py:120
↓ 4 callersFunctiondraw_uncurated_result_figure
(png, Gs, cx, cy, cw, ch, rows, lods, seed)
generate_figures.py:41
↓ 4 callersFunctionfromrgb
(x, res)
training/networks_progan.py:269
↓ 4 callersFunctionget_weight
(shape, gain=np.sqrt(2), use_wscale=False)
training/networks_progan.py:25
↓ 4 callersFunctionget_weight
(shape, gain=np.sqrt(2), use_wscale=False, lrmul=1)
training/networks_stylegan.py:135
↓ 4 callersFunctionnormalize
(v)
metrics/perceptual_path_length.py:20
↓ 3 callersFunction_downscale2d
(x, factor=2, gain=1)
training/networks_stylegan.py:70
↓ 3 callersMethod_init_fields
(self)
dnnlib/tflib/network.py:101
↓ 3 callersMethod_init_graph
(self)
dnnlib/tflib/network.py:126
↓ 3 callersFunction_upscale2d
(x, factor=2, gain=1)
training/networks_stylegan.py:51
↓ 3 callersMethodclose
Flush, close possible files, and remove stdout/stderr mirroring.
dnnlib/util.py:94
↓ 3 callersFunctiondense
(x, fmaps, gain=np.sqrt(2), use_wscale=False)
training/networks_progan.py:38
↓ 3 callersMethodget_loss_scaling_var
Get or create variable representing log2 of the current dynamic loss scaling factor.
dnnlib/tflib/optimizer.py:187
↓ 3 callersFunctionget_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
dnnlib/util.py:236
↓ 3 callersFunctionget_path_from_template
Replace tags in the given path template and return either Windows or Linux formatted path.
dnnlib/submission/submit.py:101
↓ 3 callersMethodget_random_labels_tf
(self, minibatch_size)
training/dataset.py:156
↓ 3 callersMethodprint_layers
Print a summary table of the network structure.
dnnlib/tflib/network.py:507
↓ 3 callersFunctionprob_normalize
(p)
metrics/linear_separability.py:66
↓ 3 callersMethodreset_optimizer_state
Reset internal state of the underlying optimizer.
dnnlib/tflib/optimizer.py:182
↓ 3 callersFunctionrun
Run the specified ops in the default session.
dnnlib/tflib/tfutil.py:23
↓ 2 callersFunction_create_var
Internal helper for creating autosummary accumulators.
dnnlib/tflib/autosummary.py:42
↓ 2 callersMethod_generate_labels
(self, minibatch)
training/dataset.py:222
↓ 2 callersFunction_sanitize_tf_config
(config_dict: dict = None)
dnnlib/tflib/tfutil.py:79
↓ 2 callersFunctionabsolute_name_scope
Forcefully enter the specified name scope, ignoring any surrounding scopes.
dnnlib/tflib/tfutil.py:69
↓ 2 callersMethodapply_updates
Construct training op to update the registered variables based on their gradients.
dnnlib/tflib/optimizer.py:102
↓ 2 callersFunctionblur
(x)
training/networks_stylegan.py:467
↓ 2 callersFunctionblur2d
(x, f=[1,2,1], normalize=True)
training/networks_stylegan.py:96
↓ 2 callersMethodconfigure
(self, minibatch_size, lod=0)
training/dataset.py:136
↓ 2 callersMethodconfigure
(self, minibatch_size, lod=0)
training/dataset.py:190
↓ 2 callersFunctionconvert_to_pil_image
(image, drange=[0,1])
training/misc.py:66
↓ 2 callersMethodcopy_vars_from
Copy the values of all variables from the given network, including sub-networks.
dnnlib/tflib/network.py:322
↓ 2 callersMethodflush
Flush written text to both stdout and a file, if open.
dnnlib/util.py:87
↓ 2 callersMethodget_last_update_interval
How much time passed between the previous two calls to update.
dnnlib/submission/run_context.py:86
↓ 2 callersMethodget_minibatch_tf
(self)
training/dataset.py:145
↓ 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:195
↓ 2 callersMethodget_result_str
(self)
metrics/metric_base.py:72
↓ 2 callersFunctionget_user_name
Get the current user name.
dnnlib/submission/submit.py:143
↓ 2 callersFunctiongrow
(x, res, lod)
training/networks_progan.py:225
↓ 2 callersFunctiongrow
(x, res, lod)
training/networks_stylegan.py:550
↓ 2 callersFunctionis_tf_expression
Check whether the input is a valid Tensorflow expression, i.e., Tensorflow Tensor, Variable, or Operation.
dnnlib/tflib/tfutil.py:29
↓ 2 callersFunctionlerp
(a, b, t)
training/networks_progan.py:18
↓ 2 callersFunctionlerp_clip
(a, b, t)
training/networks_progan.py:19
↓ 2 callersFunctionlocate_run_dir
(run_id_or_run_dir)
training/misc.py:92
↓ 2 callersFunctionpixel_norm
(x, epsilon=1e-8)
training/networks_progan.py:124
↓ 2 callersFunctionpixel_norm
(x, epsilon=1e-8)
training/networks_stylegan.py:239
↓ 2 callersMethodregister_gradients
Register the gradients of the given loss function with respect to the given variables. Intended to be called once per GPU.
dnnlib/tflib/optimizer.py:67
↓ 2 callersMethodreset_own_vars
Re-initialize all variables of this network, excluding sub-networks.
dnnlib/tflib/network.py:188
↓ 2 callersMethodsetup_weight_histograms
Construct summary ops to include histograms of all trainable parameters in TensorBoard.
dnnlib/tflib/network.py:536
↓ 2 callersFunctionslerp
(a, b, t)
metrics/perceptual_path_length.py:24
↓ 2 callersFunctiontraining_schedule
( cur_nimg, training_set, num_gpus, lod_initial_resolution = 4, # Image resolution use
training/training_loop.py:55
↓ 2 callersMethodwrite
Write text to stdout (and a file) and optionally flush.
dnnlib/util.py:74
↓ 1 callersMethod__enter__
(self)
dnnlib/submission/run_context.py:55
↓ 1 callersMethod__init__
(self, num_threads)
dataset_tool.py:132
↓ 1 callersFunction_create_run_dir_local
Create a new run dir with increasing ID number at the start.
dnnlib/submission/submit.py:159
↓ 1 callersMethod_evaluate
(self, Gs, num_gpus)
metrics/metric_base.py:87
↓ 1 callersMethod_generate_images
(self, minibatch, lod, shape)
training/dataset.py:219
↓ 1 callersMethod_get_cache_file_for_reals
(self, extension='pkl', **kwargs)
metrics/metric_base.py:93
↓ 1 callersFunction_get_next_run_id_local
Reads all directory names in a given directory (non-recursive) and returns the next (increasing) run id. Assumes IDs are numbers at the start of the d
dnnlib/submission/submit.py:180
next →1–100 of 280, ranked by callers