MCPcopy Create free account

hub / github.com/NVlabs/stylegan2 / functions

Functions345 in github.com/NVlabs/stylegan2

↓ 47 callersMethodget_output_for
Construct TensorFlow expression(s) for the output(s) of this network, given the input expression(s).
dnnlib/tflib/network.py:199
↓ 29 callersFunctionautosummary
Create a new autosummary. Args: name: Name to use in TensorBoard value: TensorFlow expression or python value to track
dnnlib/tflib/autosummary.py:77
↓ 27 callersMethodget
()
dnnlib/submission/run_context.py:106
↓ 26 callersMethodrun
(self, target_images)
projector.py:134
↓ 20 callersFunctionnf
(stage)
training/networks_stylegan2.py:328
↓ 20 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:63
↓ 18 callersFunctionapply_bias_act
(x, act='linear', alpha=None, gain=None, lrmul=1, bias_var='bias')
training/networks_stylegan2.py:66
↓ 16 callersFunction_shape
(tf_expr, dim_idx)
dnnlib/tflib/ops/upfirdn_2d.py:337
↓ 13 callersMethod_info
(self, *args)
projector.py:50
↓ 13 callersFunctionadd_command
(cmd, desc, example=None)
dataset_tool.py:554
↓ 13 callersMethodclone
Create a clone of this network with its own copy of the variables.
dnnlib/tflib/network.py:301
↓ 11 callersFunctionnf
(stage)
training/networks_stylegan.py:465
↓ 11 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:353
↓ 10 callersMethodadd_image
(self, img)
dataset_tool.py:65
↓ 10 callersFunctionapply_bias
(x, lrmul=1)
training/networks_stylegan.py:212
↓ 10 callersFunctionconv2d_layer
(x, fmaps, kernel, up=False, down=False, resample_kernel=None, gain=1, use_wscale=True, lrmul=1, weight_var='w
training/networks_stylegan2.py:51
↓ 9 callersMethod_report_result
(self, value, suffix='', fmt='%-10.4f')
metrics/metric_base.py:97
↓ 9 callersMethodrun
(self, network_pkl, run_dir=None, data_dir=None, dataset_args=None, mirror_augment=None, num_gpus=1, tf_config
metrics/metric_base.py:61
↓ 8 callersFunctionblock
(res, x)
training/networks_stylegan2.py:368
↓ 8 callersFunctionblock
(res, x)
training/networks_stylegan.py:516
↓ 8 callersFunctionconv2d
(x, fmaps, kernel, **kwargs)
training/networks_stylegan.py:163
↓ 8 callersMethodget_minibatch_np
(self, minibatch_size, lod=0)
training/dataset.py:139
↓ 8 callersFunctiontorgb
(res, x)
training/networks_stylegan2.py:375
↓ 7 callersMethod_report_progress
(self, pcur, pmax, status_str='')
metrics/metric_base.py:100
↓ 7 callersMethodchoose_shuffled_order
(self)
dataset_tool.py:60
↓ 7 callersFunctionfromrgb
(x, res)
training/networks_stylegan2.py:544
↓ 6 callersFunctiondense_layer
(x, fmaps, gain=1, use_wscale=True, lrmul=1, weight_var='weight')
training/networks_stylegan2.py:41
↓ 6 callersFunctionfunc
(x)
training/networks_stylegan.py:98
↓ 6 callersMethodget_random_labels_tf
(self, minibatch_size)
training/dataset.py:147
↓ 6 callersFunctionlayer
(x, layer_idx, fmaps, kernel, up=False)
training/networks_stylegan2.py:349
↓ 6 callersFunctiontorgb
(res, x)
training/networks_stylegan.py:523
↓ 5 callersMethod_get_random_labels_tf
(self, minibatch_size)
metrics/metric_base.py:139
↓ 5 callersFunction_setup_kernel
(k)
dnnlib/tflib/ops/upfirdn_2d.py:344
↓ 5 callersFunction_simple_upfirdn_2d
(x, k, up=1, down=1, pad0=0, pad1=0, data_format='NCHW', impl='cuda')
dnnlib/tflib/ops/upfirdn_2d.py:353
↓ 5 callersMethodadd_labels
(self, labels)
dataset_tool.py:90
↓ 5 callersMethodapply_updates
Construct training op to update the registered variables based on their gradients.
dnnlib/tflib/optimizer.py:154
↓ 5 callersMethodcopy_vars_from
Copy the values of all variables from the given network, including sub-networks.
dnnlib/tflib/network.py:321
↓ 5 callersFunctiondense
(x, fmaps, **kwargs)
training/networks_stylegan.py:153
↓ 5 callersFunctionerror
(msg)
dataset_tool.py:26
↓ 5 callersFunctionfromrgb
(x, res)
training/networks_stylegan.py:598
↓ 5 callersFunctionlayer_epilogue
(x, layer_idx)
training/networks_stylegan.py:489
↓ 5 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:112
↓ 5 callersFunctionupscale2d
(x, factor=2)
training/networks_stylegan.py:107
↓ 4 callersFunction_blur2d
(x, f=[1,2,1], normalize=True, flip=False, stride=1)
training/networks_stylegan.py:21
↓ 4 callersFunctionassert_tf_initialized
Check that TensorFlow session has been initialized.
dnnlib/tflib/tfutil.py:135
↓ 4 callersMethodclose
Close the context and clean up. Should only be called once.
dnnlib/submission/run_context.py:90
↓ 4 callersFunctioncset
(cur_lambda, new_cond, new_lambda)
training/networks_stylegan2.py:399
↓ 4 callersFunctioncset
(cur_lambda, new_cond, new_lambda)
training/networks_stylegan.py:547
↓ 4 callersFunctiondownscale2d
(x, factor=2)
training/networks_stylegan.py:119
↓ 4 callersMethodfind_var
Find variable by local or global name.
dnnlib/tflib/network.py:240
↓ 4 callersMethodget_cur_step
(self)
projector.py:194
↓ 4 callersFunctionget_weight
(shape, gain=np.sqrt(2), use_wscale=False, lrmul=1)
training/networks_stylegan.py:134
↓ 4 callersFunctionmodulated_conv2d_layer
(x, y, fmaps, kernel, up=False, down=False, demodulate=True, resample_kernel=None, gain=1, use_wscale=True, lr
training/networks_stylegan2.py:89
↓ 4 callersFunctionnormalize
(v)
metrics/perceptual_path_length.py:19
↓ 4 callersMethodreset_optimizer_state
Reset internal state of the underlying optimizer.
dnnlib/tflib/optimizer.py:264
↓ 3 callersFunction_downscale2d
(x, factor=2, gain=1)
training/networks_stylegan.py:69
↓ 3 callersMethod_init_fields
(self)
dnnlib/tflib/network.py:100
↓ 3 callersMethod_init_graph
(self)
dnnlib/tflib/network.py:125
↓ 3 callersMethod_reset
(self, network_pkl=None, run_dir=None, data_dir=None, dataset_args=None, mirror_augment=None)
metrics/metric_base.py:37
↓ 3 callersFunction_upscale2d
(x, factor=2, gain=1)
training/networks_stylegan.py:50
↓ 3 callersFunctionget_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
dnnlib/util.py:235
↓ 3 callersFunctionget_path_from_template
Replace tags in the given path template and return either Windows or Linux formatted path.
dnnlib/submission/submit.py:116
↓ 3 callersMethodget_result_str
(self)
metrics/metric_base.py:79
↓ 3 callersFunctionget_weight
(shape, gain=1, use_wscale=True, lrmul=1, weight_var='weight')
training/networks_stylegan2.py:22
↓ 3 callersFunctionprob_normalize
(p)
metrics/linear_separability.py:65
↓ 3 callersFunctionrun
Run the specified ops in the default session.
dnnlib/tflib/tfutil.py:28
↓ 3 callersMethodstart
(self, target_images)
projector.py:147
↓ 3 callersFunctiontraining_schedule
( cur_nimg, training_set, lod_initial_resolution = None, # Image resolution used at the begin
training/training_loop.py:47
↓ 3 callersFunctionupsample_2d
r"""Upsample a batch of 2D images with the given filter. Accepts a batch of 2D images of the shape `[N, C, H, W]` or `[N, H, W, C]` and upsam
dnnlib/tflib/ops/upfirdn_2d.py:169
↓ 2 callersMethod_broadcast_fallback
Sum gradients across devices using TensorFlow collective ops (slow fallback path).
dnnlib/tflib/optimizer.py:297
↓ 2 callersFunction_create_var
Internal helper for creating autosummary accumulators.
dnnlib/tflib/autosummary.py:45
↓ 2 callersMethod_get_cache_file_for_reals
(self, extension='pkl', **kwargs)
metrics/metric_base.py:110
↓ 2 callersMethod_get_dataset_obj
(self)
metrics/metric_base.py:119
↓ 2 callersMethod_get_device
Get internal state for the given TensorFlow device.
dnnlib/tflib/optimizer.py:82
↓ 2 callersFunction_get_plugin
()
dnnlib/tflib/ops/upfirdn_2d.py:14
↓ 2 callersMethod_iterate_reals
(self, minibatch_size)
metrics/metric_base.py:124
↓ 2 callersFunction_prepare_nvcc_cli
(opts)
dnnlib/tflib/custom_ops.py:63
↓ 2 callersFunction_run_cmd
(cmd)
dnnlib/tflib/custom_ops.py:56
↓ 2 callersFunction_sanitize_tf_config
(config_dict: dict = None)
dnnlib/tflib/tfutil.py:84
↓ 2 callersFunctionabsolute_name_scope
Forcefully enter the specified name scope, ignoring any surrounding scopes.
dnnlib/tflib/tfutil.py:74
↓ 2 callersFunctionblur
(x)
training/networks_stylegan.py:466
↓ 2 callersFunctionblur2d
(x, f=[1,2,1], normalize=True)
training/networks_stylegan.py:95
↓ 2 callersMethodclose
(self)
training/dataset.py:122
↓ 2 callersMethodclose
Flush, close possible files, and remove stdout/stderr mirroring.
dnnlib/util.py:93
↓ 2 callersMethodconfigure
(self, minibatch_size, lod=0)
training/dataset.py:126
↓ 2 callersFunctionconv_downsample_2d
r"""Fused `tf.nn.conv2d()` followed by `downsample_2d()`. Padding is performed only once at the beginning, not between the operations. The fu
dnnlib/tflib/ops/upfirdn_2d.py:296
↓ 2 callersFunctiondownsample_2d
r"""Downsample a batch of 2D images with the given filter. Accepts a batch of 2D images of the shape `[N, C, H, W]` or `[N, H, W, C]` and dow
dnnlib/tflib/ops/upfirdn_2d.py:202
↓ 2 callersMethodevaluate
Evaluate if new feature vectors are in the estimated manifold.
metrics/precision_recall.py:96
↓ 2 callersMethodflush
Flush written text to both stdout and a file, if open.
dnnlib/util.py:86
↓ 2 callersFunctionfunc_y
(x, b)
dnnlib/tflib/ops/fused_bias_act.py:126
↓ 2 callersMethodget_images
(self)
projector.py:203
↓ 2 callersMethodget_last_update_interval
How much time passed between the previous two calls to update.
dnnlib/submission/run_context.py:86
↓ 2 callersMethodget_loss_scaling_var
Get or create variable representing log2 of the current dynamic loss scaling factor.
dnnlib/tflib/optimizer.py:269
↓ 2 callersMethodget_minibatch_tf
(self)
training/dataset.py:135
↓ 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:194
↓ 2 callersFunctionget_user_name
Get the current user name.
dnnlib/submission/submit.py:157
↓ 2 callersFunctiongrad2_d_dy
(d_dx, d_db, x, y)
dnnlib/tflib/ops/fused_bias_act.py:149
↓ 2 callersFunctiongrad_db
(dx)
dnnlib/tflib/ops/fused_bias_act.py:137
↓ 2 callersFunctiongrad_dx
(dy, x, y)
dnnlib/tflib/ops/fused_bias_act.py:132
↓ 2 callersFunctiongrow
(x, res, lod)
training/networks_stylegan2.py:401
next →1–100 of 345, ranked by callers