Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/NVlabs/stylegan2-ada
/ functions
Functions
336 in github.com/NVlabs/stylegan2-ada
⨍
Functions
336
◇
Types & classes
25
↓ 2 callers
Function
translate_2d
(tx, ty)
training/augment.py:186
↓ 2 callers
Function
upsample_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:176
↓ 1 callers
Method
__enter__
(self)
dnnlib/util.py:72
↓ 1 callers
Method
__init__
(self, num_threads)
dataset_tool.py:237
↓ 1 callers
Method
_broadcast_nccl
Sum gradients across devices using NCCL ops (fast path).
dnnlib/tflib/optimizer.py:289
↓ 1 callers
Method
_evaluate
(self, **_kwargs)
metrics/metric_base.py:90
↓ 1 callers
Function
_find_compiler_bindir
()
dnnlib/tflib/custom_ops.py:35
↓ 1 callers
Function
_fused_bias_act_ref
Slow reference implementation of `fused_bias_act()` using standard TensorFlow ops.
dnnlib/tflib/ops/fused_bias_act.py:76
↓ 1 callers
Function
_get_compute_cap
(device)
dnnlib/tflib/custom_ops.py:50
↓ 1 callers
Function
_get_cuda_gpu_arch_string
()
dnnlib/tflib/custom_ops.py:57
↓ 1 callers
Function
_get_plugin
()
dnnlib/tflib/ops/upfirdn_2d.py:16
↓ 1 callers
Function
_get_plugin
()
dnnlib/tflib/ops/fused_bias_act.py:17
↓ 1 callers
Function
_handle_legacy_output_transforms
(output_transform, dynamic_kwargs)
dnnlib/tflib/network.py:746
↓ 1 callers
Method
_increment_acc
(self, name, expr)
training/augment.py:138
↓ 1 callers
Method
_read_and_decay_acc
(self, name, nimg_delta)
training/augment.py:151
↓ 1 callers
Method
_set_strength
(self, strength)
training/augment.py:132
↓ 1 callers
Method
add_task
(self, func, args=())
dataset_tool.py:247
↓ 1 callers
Method
apply
(self, images, labels, enable=True)
training/augment.py:90
↓ 1 callers
Method
apply_gradients
(self, grads_and_vars)
dnnlib/tflib/optimizer.py:344
↓ 1 callers
Method
apply_loss_scaling
Apply dynamic loss scaling for the given expression.
dnnlib/tflib/optimizer.py:275
↓ 1 callers
Function
apply_spectral_norm
(w, state_var='sn', iterations=1, eps=1e-8)
training/networks.py:168
↓ 1 callers
Function
batch_pairwise_distances
Compute pairwise distances between two batches of feature vectors.
metrics/precision_recall.py:23
↓ 1 callers
Function
calc_metrics
(network_pkl, metric_names, metricdata, mirror, gpus)
calc_metrics.py:27
↓ 1 callers
Function
call_func_by_name
Finds the python object with the given name and calls it as a function.
dnnlib/util.py:276
↓ 1 callers
Method
close
(self)
dataset_tool.py:60
↓ 1 callers
Method
compute_gradients
(self, loss, var_list, gate_gradients=tf.train.Optimizer.GATE_NONE)
dnnlib/tflib/optimizer.py:340
↓ 1 callers
Function
compute_kid
(feat_real, feat_fake, num_subsets=100, max_subset_size=1000)
metrics/kernel_inception_distance.py:23
↓ 1 callers
Function
conditional_entropy
(p)
metrics/linear_separability.py:98
↓ 1 callers
Method
configure
(self, dataset_args={}, run_dir=None, progress_fn=None)
metrics/metric_base.py:43
↓ 1 callers
Function
conv_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:313
↓ 1 callers
Method
convert
Create new network with the given parameters, and copy all variables from this network.
dnnlib/tflib/network.py:518
↓ 1 callers
Method
copy_own_vars_from
Copy the values of all variables from the given network, excluding sub-networks.
dnnlib/tflib/network.py:460
↓ 1 callers
Function
create_session
Create tf.Session based on config dict.
dnnlib/tflib/tfutil.py:151
↓ 1 callers
Function
downsample
(y)
training/networks.py:584
↓ 1 callers
Function
entropy
(p)
metrics/linear_separability.py:88
↓ 1 callers
Function
execute_cmdline
(argv)
dataset_tool.py:861
↓ 1 callers
Function
finalize_autosummaries
Create the necessary ops to include autosummaries in TensorBoard report. Note: This should be done only once per graph.
dnnlib/tflib/autosummary.py:120
↓ 1 callers
Method
finish
(self)
dataset_tool.py:260
↓ 1 callers
Function
func_nonzero_2nd_grad
(x, b)
dnnlib/tflib/ops/fused_bias_act.py:191
↓ 1 callers
Function
func_zero_2nd_grad
(x, b)
dnnlib/tflib/ops/fused_bias_act.py:177
↓ 1 callers
Function
fused_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:36
↓ 1 callers
Function
generate_images
(network_pkl, seeds, truncation_psi, outdir, class_idx, dlatents_npz)
generate.py:24
↓ 1 callers
Function
get_obj_by_name
Finds the python object with the given name.
dnnlib/util.py:270
↓ 1 callers
Method
get_result
(self, func)
dataset_tool.py:253
↓ 1 callers
Method
get_result_str
(self)
metrics/metric_base.py:77
↓ 1 callers
Method
get_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:393
↓ 1 callers
Function
grad2_d_x
(d_dx, d_db, x, y)
dnnlib/tflib/ops/fused_bias_act.py:170
↓ 1 callers
Function
grad_impl
(dy, x)
dnnlib/tflib/ops/fused_bias_act.py:195
↓ 1 callers
Method
init_validation_set
(self, D_gpus, training_set)
training/augment.py:67
↓ 1 callers
Function
is_top_level_function
Determine whether the given object is a top-level function, i.e., defined at module scope using 'def'.
dnnlib/util.py:295
↓ 1 callers
Function
is_url
Determine whether the given object is a valid URL string.
dnnlib/util.py:361
↓ 1 callers
Function
knn_precision_recall_features
Calculates k-NN precision and recall for two sets of feature vectors.
metrics/precision_recall.py:141
↓ 1 callers
Method
list_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:651
↓ 1 callers
Method
list_ops
(self)
dnnlib/tflib/network.py:642
↓ 1 callers
Method
load_validation_set_np
(self)
training/dataset.py:183
↓ 1 callers
Function
main
()
train.py:509
↓ 1 callers
Function
main
()
calc_metrics.py:138
↓ 1 callers
Function
main
()
style_mixing.py:98
↓ 1 callers
Function
main
()
generate.py:100
↓ 1 callers
Function
main
()
projector.py:270
↓ 1 callers
Function
make_cache_dir_path
(*paths: str)
dnnlib/util.py:121
↓ 1 callers
Function
minibatch_stddev_layer
(x, group_size=None, num_new_features=1)
training/networks.py:142
↓ 1 callers
Function
mutual_information
(p)
metrics/linear_separability.py:74
↓ 1 callers
Method
parse_tfrecord_np
(record)
training/dataset.py:215
↓ 1 callers
Function
project
(network_pkl: str, target_fname: str, outdir: str, save_video: bool, seed: int)
projector.py:205
↓ 1 callers
Method
report_stat
(self, name, expr)
training/augment.py:103
↓ 1 callers
Function
rotate_2d
(theta)
training/augment.py:212
↓ 1 callers
Function
rotate_3d
(v, theta)
training/augment.py:218
↓ 1 callers
Function
run_training
(outdir, seed, dry_run, **hyperparam_options)
train.py:413
↓ 1 callers
Method
run_validation
(self, minibatch_size)
training/augment.py:126
↓ 1 callers
Function
scale_3d
(sx, sy, sz)
training/augment.py:205
↓ 1 callers
Method
set_network
(self, Gs, dtype='float16')
projector.py:62
↓ 1 callers
Function
set_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:204
↓ 1 callers
Method
setup_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:528
↓ 1 callers
Function
setup_snapshot_image_grid
(training_set)
training/training_loop.py:27
↓ 1 callers
Function
setup_training_options
( # General options (not included in desc). gpus = None, # Number of GPUs: <int>, default = 1 gp
train.py:31
↓ 1 callers
Method
step
(self)
projector.py:163
↓ 1 callers
Function
style_mixing_example
(network_pkl, row_seeds, col_seeds, truncation_psi, col_styles, outdir, minibatch_size=4)
style_mixing.py:24
↓ 1 callers
Function
translate_3d
(tx, ty, tz)
training/augment.py:192
↓ 1 callers
Method
tune
(self, nimg_delta)
training/augment.py:108
↓ 1 callers
Method
undo_loss_scaling
Undo the effect of dynamic loss scaling for the given expression.
dnnlib/tflib/optimizer.py:282
↓ 1 callers
Method
update_autosummaries
(self)
metrics/metric_base.py:86
↓ 1 callers
Function
upsample
(y)
training/networks.py:417
↓ 1 callers
Function
upsample_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:245
↓ 1 callers
Method
variables
(self)
dnnlib/tflib/optimizer.py:337
Function
D_main
( images_in, # First input: Images [minibatch, channel, height, width]. label
training/networks.py:459
Function
G_main
( latents_in, # First input: Latent vectors (Z) [minibatch, latent_siz
training/networks.py:188
Function
G_mapping
( latents_in, # First input: Latent vectors (Z) [minibatch, latent_size].
training/networks.py:283
Function
G_synthesis
( dlatents_in, # Input: Disentangled latents (W) [minibatch, num_layers, dlatent_si
training/networks.py:343
Method
__delattr__
(self, name: str)
dnnlib/util.py:52
Method
__enter__
(self)
dataset_tool.py:107
Method
__enter__
(self)
dataset_tool.py:203
Method
__enter__
(self)
dataset_tool.py:264
Method
__exit__
(self, *args)
dataset_tool.py:110
Method
__exit__
(self, *args)
dataset_tool.py:206
Method
__exit__
(self, *excinfo)
dataset_tool.py:267
Method
__exit__
(self, exc_type: Any, exc_value: Any, traceback: Any)
dnnlib/util.py:75
Method
__getattr__
(self, name: str)
dnnlib/util.py:43
Method
__getstate__
Pickle export.
dnnlib/tflib/network.py:403
Method
__init__
(self, tfrecord_dir, expected_images, print_progress=True, progress_interval=10, tfr_prefix=None)
dataset_tool.py:39
← previous
next →
101–200 of 336, ranked by callers