MCPcopy Create free account

hub / github.com/NVlabs/stylegan2 / functions

Functions345 in github.com/NVlabs/stylegan2

↓ 2 callersFunctiongrow
(x, res, lod)
training/networks_stylegan.py:549
↓ 2 callersFunctionis_tf_expression
Check whether the input is a valid Tensorflow expression, i.e., Tensorflow Tensor, Variable, or Operation.
dnnlib/tflib/tfutil.py:34
↓ 2 callersFunctionminibatch_stddev_layer
(x, group_size=4, num_new_features=1)
training/networks_stylegan2.py:132
↓ 2 callersFunctionnaive_downsample_2d
(x, factor=2)
training/networks_stylegan2.py:80
↓ 2 callersFunctionnaive_upsample_2d
(x, factor=2)
training/networks_stylegan2.py:73
↓ 2 callersMethodpairwise_distances
Evaluate pairwise distances between two batches of feature vectors.
metrics/precision_recall.py:55
↓ 2 callersFunctionpixel_norm
(x, epsilon=1e-8)
training/networks_stylegan.py:238
↓ 2 callersMethodprint_layers
Print a summary table of the network structure.
dnnlib/tflib/network.py:506
↓ 2 callersFunctionproject_image
(proj, targets, png_prefix, num_snapshots)
run_projector.py:21
↓ 2 callersMethodreset_own_vars
Re-initialize all variables of this network, excluding sub-networks.
dnnlib/tflib/network.py:187
↓ 2 callersMethodrun
(self, *args, **kwargs)
metrics/metric_base.py:149
↓ 2 callersMethodset_network
(self, Gs, minibatch_size=1)
projector.py:54
↓ 2 callersMethodsetup_weight_histograms
Construct summary ops to include histograms of all trainable parameters in TensorBoard.
dnnlib/tflib/network.py:535
↓ 2 callersFunctionslerp
(a, b, t)
metrics/perceptual_path_length.py:23
↓ 2 callersMethodstep
(self)
projector.py:167
↓ 2 callersFunctionupsample_conv_2d
r"""Fused `upsample_2d()` followed by `tf.nn.conv2d()`. Padding is performed only once at the beginning, not between the operations. The fuse
dnnlib/tflib/ops/upfirdn_2d.py:234
↓ 2 callersMethodwrite
Write text to stdout (and a file) and optionally flush.
dnnlib/util.py:73
↓ 1 callersMethod__enter__
(self)
dnnlib/submission/run_context.py:57
↓ 1 callersMethod__init__
(self, num_threads)
dataset_tool.py:131
↓ 1 callersMethod_broadcast_nccl
Sum gradients across devices using NCCL ops (fast path).
dnnlib/tflib/optimizer.py:287
↓ 1 callersFunction_create_run_dir_local
Create a new run dir with increasing ID number at the start.
dnnlib/submission/submit.py:192
↓ 1 callersMethod_evaluate
(self, Gs, Gs_kwargs, num_gpus)
metrics/metric_base.py:94
↓ 1 callersFunction_find_compiler_bindir
()
dnnlib/tflib/custom_ops.py:36
↓ 1 callersFunction_fused_bias_act_ref
Slow reference implementation of `fused_bias_act()` using standard TensorFlow ops.
dnnlib/tflib/ops/fused_bias_act.py:72
↓ 1 callersFunction_get_compute_cap
(device)
dnnlib/tflib/custom_ops.py:42
↓ 1 callersFunction_get_cuda_gpu_arch_string
()
dnnlib/tflib/custom_ops.py:49
↓ 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:211
↓ 1 callersFunction_get_plugin
()
dnnlib/tflib/ops/fused_bias_act.py:15
↓ 1 callersFunction_handle_legacy_output_transforms
(output_transform, dynamic_kwargs)
dnnlib/tflib/network.py:555
↓ 1 callersFunction_populate_run_dir
Copy all necessary files into the run dir. Assumes that the dir exists, is local, and is writable.
dnnlib/submission/submit.py:227
↓ 1 callersMethodadd_task
(self, func, args=())
dataset_tool.py:141
↓ 1 callersFunctionadjust_dynamic_range
(data, drange_in, drange_out)
training/misc.py:36
↓ 1 callersMethodapply_gradients
(self, grads_and_vars)
dnnlib/tflib/optimizer.py:342
↓ 1 callersMethodapply_loss_scaling
Apply dynamic loss scaling for the given expression.
dnnlib/tflib/optimizer.py:273
↓ 1 callersFunctionapply_noise
(x, noise_var=None, randomize_noise=True)
training/networks_stylegan.py:269
↓ 1 callersFunctionbatch_pairwise_distances
Compute pairwise distances between two batches of feature vectors.
metrics/precision_recall.py:20
↓ 1 callersMethodclose
(self)
dataset_tool.py:50
↓ 1 callersMethodclose
(self)
metrics/metric_base.py:34
↓ 1 callersMethodcompute_gradients
(self, loss, var_list, gate_gradients=tf.train.Optimizer.GATE_NONE)
dnnlib/tflib/optimizer.py:338
↓ 1 callersFunctionconditional_entropy
(p)
metrics/linear_separability.py:94
↓ 1 callersFunctionconv2d_downscale2d
(x, fmaps, kernel, fused_scale='auto', **kwargs)
training/networks_stylegan.py:192
↓ 1 callersFunctionconvert_path
Convert a normal path to template and the convert it back to a normal path with given path type.
dnnlib/submission/submit.py:144
↓ 1 callersFunctionconvert_to_pil_image
(image, drange=[0,1])
training/misc.py:60
↓ 1 callersFunctioncreate_image_grid
(images, grid_size=None)
training/misc.py:43
↓ 1 callersFunctioncreate_session
Create tf.Session based on config dict.
dnnlib/tflib/tfutil.py:141
↓ 1 callersFunctiondownsample
(y)
training/networks_stylegan2.py:658
↓ 1 callersFunctionentropy
(p)
metrics/linear_separability.py:84
↓ 1 callersFunctionexecute_cmdline
(argv)
dataset_tool.py:545
↓ 1 callersFunctionfinalize_autosummaries
Create the necessary ops to include autosummaries in TensorBoard report. Note: This should be done only once per graph.
dnnlib/tflib/autosummary.py:118
↓ 1 callersMethodfinalize_submit_config
(self, submit_config, host_run_dir)
dnnlib/submission/internal/local.py:15
↓ 1 callersMethodfinish
(self)
dataset_tool.py:154
↓ 1 callersFunctionfunc
(x)
dnnlib/tflib/ops/upfirdn_2d.py:131
↓ 1 callersFunctionfunc_nonzero_2nd_grad
(x, b)
dnnlib/tflib/ops/fused_bias_act.py:176
↓ 1 callersFunctionfunc_zero_2nd_grad
(x, b)
dnnlib/tflib/ops/fused_bias_act.py:162
↓ 1 callersFunctionfused_bias_act
r"""Fused bias and activation function. Adds bias `b` to activation tensor `x`, evaluates activation function `act`, and scales the result by
dnnlib/tflib/ops/fused_bias_act.py:34
↓ 1 callersMethodget_dlatents
(self)
projector.py:197
↓ 1 callersMethodget_noises
(self)
projector.py:200
↓ 1 callersFunctionget_obj_by_name
Finds the python object with the given name.
dnnlib/util.py:245
↓ 1 callersFunctionget_path_or_url
(path_or_gdrive_path)
pretrained_networks.py:57
↓ 1 callersMethodget_result
(self, func)
dataset_tool.py:147
↓ 1 callersFunctionget_template_from_path
Convert a normal path back to its template representation.
dnnlib/submission/submit.py:138
↓ 1 callersMethodget_time_since_last_update
How much time has passed since the last call to update.
dnnlib/submission/run_context.py:82
↓ 1 callersMethodget_time_since_start
How much time has passed since the creation of the context.
dnnlib/submission/run_context.py:78
↓ 1 callersMethodget_var
Get the value of a given variable as NumPy array. Note: This method is very inefficient -- prefer to use tflib.run(list_of_vars) whenever poss
dnnlib/tflib/network.py:245
↓ 1 callersFunctiongrad2_d_x
(d_dx, d_db, x, y)
dnnlib/tflib/ops/fused_bias_act.py:154
↓ 1 callersFunctiongrad_impl
(dy, x)
dnnlib/tflib/ops/fused_bias_act.py:180
↓ 1 callersFunctioninstance_norm
(x, epsilon=1e-8)
training/networks_stylegan.py:246
↓ 1 callersFunctionis_top_level_function
Determine whether the given object is a top-level function, i.e., defined at module scope using 'def'.
dnnlib/util.py:265
↓ 1 callersFunctionis_url
Determine whether the given object is a valid URL string.
dnnlib/util.py:328
↓ 1 callersFunctionknn_precision_recall_features
Calculates k-NN precision and recall for two sets of feature vectors.
metrics/precision_recall.py:138
↓ 1 callersMethodlist_layers
Returns a list of (layer_name, output_expr, trainable_vars) tuples corresponding to individual layers of the network. Mainly intended to be us
dnnlib/tflib/network.py:463
↓ 1 callersMethodlist_ops
(self)
dnnlib/tflib/network.py:455
↓ 1 callersFunctionmain
()
run_training.py:156
↓ 1 callersFunctionmain
()
run_metrics.py:52
↓ 1 callersFunctionmain
()
run_projector.py:97
↓ 1 callersFunctionmain
()
run_generator.py:119
↓ 1 callersFunctionminibatch_stddev_layer
(x, group_size=4, num_new_features=1)
training/networks_stylegan.py:282
↓ 1 callersFunctionmutual_information
(p)
metrics/linear_separability.py:70
↓ 1 callersFunctionopen_file_or_url
(file_or_url)
training/misc.py:20
↓ 1 callersMethodparse_tfrecord_np
(record)
training/dataset.py:171
↓ 1 callersFunctionprocess_reals
(x, labels, lod, mirror_augment, drange_data, drange_net)
training/training_loop.py:22
↓ 1 callersFunctionrun
(dataset, data_dir, result_dir, config_id, num_gpus, total_kimg, gamma, mirror_augment, metrics)
run_training.py:36
↓ 1 callersFunctionrun_wrapper
Wrap the actual run function call for handling logging, exceptions, typing, etc.
dnnlib/submission/submit.py:256
↓ 1 callersFunctionset_vars
Set the values of given tf.Variables. Equivalent to the following, but more efficient and does not bloat the tf graph: tflib.run([tf.assign(v
dnnlib/tflib/tfutil.py:194
↓ 1 callersMethodsetup_as_moving_average_of
Construct a TensorFlow op that updates the variables of this network to be slightly closer to those of the given network.
dnnlib/tflib/network.py:341
↓ 1 callersMethodshould_stop
Tell whether a stopping condition has been triggered one way or another.
dnnlib/submission/run_context.py:74
↓ 1 callersFunctionstyle_mod
(x, dlatent, **kwargs)
training/networks_stylegan.py:260
↓ 1 callersMethodsubmit
(self, submit_config, host_run_dir)
dnnlib/submission/internal/local.py:19
↓ 1 callersMethodundo_loss_scaling
Undo the effect of dynamic loss scaling for the given expression.
dnnlib/tflib/optimizer.py:280
↓ 1 callersMethodupdate_autosummaries
(self)
metrics/metric_base.py:90
↓ 1 callersMethodupdate_autosummaries
(self)
metrics/metric_base.py:156
↓ 1 callersFunctionupfirdn_2d
r"""Pad, upsample, FIR filter, and downsample a batch of 2D images. Accepts a batch of 2D images of the shape `[majorDim, inH, inW, minorDim]`
dnnlib/tflib/ops/upfirdn_2d.py:19
↓ 1 callersFunctionupsample
(y)
training/networks_stylegan2.py:476
↓ 1 callersFunctionupscale2d_conv2d
(x, fmaps, kernel, fused_scale='auto', **kwargs)
training/networks_stylegan.py:173
↓ 1 callersMethodvariables
(self)
dnnlib/tflib/optimizer.py:335
FunctionD_basic
( images_in, # First input: Images [minibatch, channel, height, width]. label
training/networks_stylegan.py:563
FunctionD_logistic
(G, D, opt, training_set, minibatch_size, reals, labels)
training/loss.py:36
FunctionD_logistic_r1
(G, D, opt, training_set, minibatch_size, reals, labels, gamma=10.0)
training/loss.py:52
FunctionD_logistic_r2
(G, D, opt, training_set, minibatch_size, reals, labels, gamma=10.0)
training/loss.py:70
FunctionD_stylegan
( images_in, # First input: Images [minibatch, channel, height, width]. label
training/networks_stylegan2.py:512
← previousnext →101–200 of 345, ranked by callers