MCPcopy Create free account

hub / github.com/NVlabs/stylegan2-ada / functions

Functions336 in github.com/NVlabs/stylegan2-ada

Method__init__
(self, h5_filename, resolution, channels, compress=False, expected_images=0, print_progress=True, progress_int
dataset_tool.py:117
Method__init__
(self)
dataset_tool.py:212
Method__init__
(self, task_queue)
dataset_tool.py:219
Method__init__
(self)
projector.py:25
Method__init__
(self, name, force_dataset_args={}, force_G_kwargs={})
metrics/metric_base.py:26
Method__init__
(self, num_features, num_gpus)
metrics/precision_recall.py:43
Method__init__
Find an estimate of the manifold of given feature vectors.
metrics/precision_recall.py:66
Method__init__
(self, max_reals, num_fakes, nhood_size, minibatch_per_gpu, row_batch_size, col_batch_size, **kwargs)
metrics/precision_recall.py:175
Method__init__
(self, num_images, num_splits, minibatch_per_gpu, **kwargs)
metrics/inception_score.py:23
Method__init__
(self, num_samples, epsilon, space, sampling, crop, minibatch_per_gpu, **kwargs)
metrics/perceptual_path_length.py:39
Method__init__
(self, num_samples, num_keep, attrib_indices, minibatch_per_gpu, **kwargs)
metrics/linear_separability.py:108
Method__init__
(self, max_reals, num_fakes, minibatch_per_gpu, use_cached_real_stats=True, **kwargs)
metrics/frechet_inception_distance.py:25
Method__init__
(self, max_reals, num_fakes, minibatch_per_gpu, use_cached_real_stats=True, **kwargs)
metrics/kernel_inception_distance.py:38
Method__init__
(self, tfrecord_dir, # Directory containing a collection of tfrecords files. res
training/dataset.py:21
Method__init__
(self, apply_func = None, # Function representing the augmentation pipeline. Can be a fully-q
training/augment.py:28
Method__init__
(self, file_name: str = None, file_mode: str = "w", should_flush: bool = True)
dnnlib/util.py:59
Method__init__
(self, name: str = None, func_name: Any = None, **static_kwargs)
dnnlib/tflib/network.py:60
Method__init__
(self, name: str = "Train", # Name string that will app
dnnlib/tflib/optimizer.py:40
Method__init__
(self, name="Adam", learning_rate=0.001, beta1=0.9, beta2=0.999, epsilon=1e-8)
dnnlib/tflib/optimizer.py:329
Method__init__
(self, k, gain=1)
dnnlib/tflib/ops/upfirdn_2d.py:371
Method__setattr__
(self, name: str, value: Any)
dnnlib/util.py:49
Method__setstate__
Pickle import.
dnnlib/tflib/network.py:419
Method_evaluate
(self, Gs, G_kwargs, num_gpus, **_kwargs)
metrics/precision_recall.py:184
Method_evaluate
(self, Gs, G_kwargs, num_gpus, **_kwargs)
metrics/inception_score.py:29
Method_evaluate
(self, Gs, G_kwargs, num_gpus, **_kwargs)
metrics/perceptual_path_length.py:50
Method_evaluate
(self, Gs, G_kwargs, num_gpus, **_kwargs)
metrics/linear_separability.py:116
Method_evaluate
(self, Gs, G_kwargs, num_gpus, **_kwargs)
metrics/frechet_inception_distance.py:32
Method_evaluate
(self, Gs, G_kwargs, num_gpus, **_kwargs)
metrics/kernel_inception_distance.py:45
Function_fused_bias_act_cuda
Fast CUDA implementation of `fused_bias_act()` using custom ops.
dnnlib/tflib/ops/fused_bias_act.py:110
Function_legacy_output_transform_func
(*expr, out_mul=1.0, out_add=0.0, out_shrink=1, out_dtype=None)
dnnlib/tflib/network.py:766
Function_parse_comma_sep
(s)
train.py:464
Function_parse_comma_sep
(s)
calc_metrics.py:97
Function_parse_num_range
Accept either a comma separated list of numbers 'a,b,c' or a range 'a-c' and return as a list of ints.
style_mixing.py:78
Function_parse_num_range
Accept either a comma separated list of numbers 'a,b,c' or a range 'a-c' and return as a list of ints.
generate.py:67
Function_str_to_bool
(v)
train.py:455
Function_str_to_bool
(v)
calc_metrics.py:88
Function_str_to_bool
(v)
projector.py:251
Function_upfirdn_2d_cuda
Fast CUDA implementation of `upfirdn_2d()` using custom ops.
dnnlib/tflib/ops/upfirdn_2d.py:107
Function_upfirdn_2d_ref
Slow reference implementation of `upfirdn_2d()` using standard TensorFlow ops.
dnnlib/tflib/ops/upfirdn_2d.py:68
Functionabsolute_variable_scope
Forcefully enter the specified variable scope, ignoring any surrounding scopes.
dnnlib/tflib/tfutil.py:88
Methodadd_image
(self, img)
dataset_tool.py:75
Methodadd_labels
(self, labels)
dataset_tool.py:100
Functionask_yes_no
Ask the user the question until the user inputs a valid answer.
dnnlib/util.py:150
Functionaugment_pipeline
( images, # Input images: NCHW, float32, dynamic range [-1,+1]. labels,
training/augment.py:265
Functioncmethods
(G, D, aug, fake_labels, real_images, real_labels, r1_gamma=10, r2_gamma=0, pl_minibatch_shrink=2, pl_
training/loss.py:161
Functioncompare
(tfrecord_dir_a, tfrecord_dir_b, ignore_labels)
dataset_tool.py:400
Methodcomponents
EasyDict of sub-networks created by the build func.
dnnlib/tflib/network.py:205
Functionconstruct_class_by_name
Finds the python class with the given name and constructs it with the given arguments.
dnnlib/util.py:284
Functionconvert_images_from_uint8
Convert a minibatch of images from uint8 to float32 with configurable dynamic range. Can be used as an input transformation for Network.run().
dnnlib/tflib/tfutil.py:240
Functionconvert_images_to_uint8
Convert a minibatch of images from float32 to uint8 with configurable dynamic range. Can be used as an output transformation for Network.run().
dnnlib/tflib/tfutil.py:250
Functionconvert_to_hdf5
(hdf5_filename, tfrecord_dir, compress)
dataset_tool.py:696
Functioncopy_files_and_create_dirs
Takes in a list of tuples of (src, dst) paths and copies files. Will create all necessary directories.
dnnlib/util.py:345
Methodcopy_trainables_from
Copy the values of all trainable variables from the given network, including sub-networks.
dnnlib/tflib/network.py:513
Functioncreate_celeba
(tfrecord_dir, celeba_dir, cx=89, cy=121)
dataset_tool.py:634
Functioncreate_cifar10
(tfrecord_dir, cifar10_dir, ignore_labels)
dataset_tool.py:478
Functioncreate_cifar100
(tfrecord_dir, cifar100_dir)
dataset_tool.py:507
Functioncreate_from_hdf5
(tfrecord_dir, hdf5_filename, shuffle)
dataset_tool.py:681
Functioncreate_from_images
(tfrecord_dir, image_dir, shuffle)
dataset_tool.py:653
Functioncreate_lsun
(tfrecord_dir, lmdb_dir, resolution=256, max_images=None)
dataset_tool.py:556
Functioncreate_lsun_wide
(tfrecord_dir, lmdb_dir, width=512, height=384, max_images=None)
dataset_tool.py:589
Functioncreate_mnist
(tfrecord_dir, mnist_dir)
dataset_tool.py:437
Functioncreate_mnistrgb
(tfrecord_dir, mnist_dir, num_images=1000000, random_seed=123)
dataset_tool.py:461
Functioncreate_svhn
(tfrecord_dir, svhn_dir)
dataset_tool.py:529
Functioncreate_var_with_large_initial_value
Create tf.Variable with large initial value without bloating the tf graph.
dnnlib/tflib/tfutil.py:230
Methodcur_step
(self)
projector.py:184
Functiondisplay
(tfrecord_dir)
dataset_tool.py:349
Methoddlatents
(self)
projector.py:188
Functionerfinv
Inverse of the error function.
dnnlib/tflib/tfutil.py:64
Functionexp2
Exponent in base 2.
dnnlib/tflib/tfutil.py:58
Functionextract
(tfrecord_dir, output_dir)
dataset_tool.py:374
Functionextract_brecahad_crops
(brecahad_dir, output_dir, cropsize=256)
dataset_tool.py:836
Functionfilter_2d
r"""Filter a batch of 2D images with the given FIR filter. Accepts a batch of 2D images of the shape `[N, C, H, W]` or `[N, H, W, C]` and fil
dnnlib/tflib/ops/upfirdn_2d.py:147
Functionflatten
Shortcut function for flattening a tensor.
dnnlib/tflib/tfutil.py:46
Functionformat_time
Convert the seconds to human readable string with days, hours, minutes and seconds.
dnnlib/util.py:136
Functionget_dtype_and_ctype
Given a type name string (or an object having a __name__ attribute), return matching Numpy and ctypes types that have the same size in bytes.
dnnlib/util.py:184
Functionget_module_dir_by_obj_name
Get the directory path of the module containing the given object name.
dnnlib/util.py:289
Functionget_plugin
(cuda_file, extra_nvcc_options=[])
dnnlib/tflib/custom_ops.py:95
Methodget_random_labels_np
(self, minibatch_size)
training/dataset.py:177
Functionget_top_level_function_name
Return the fully-qualified name of a top-level function.
dnnlib/util.py:300
Methodget_var_local_name
Get the local name of a given variable, without any surrounding name scopes.
dnnlib/tflib/network.py:382
Functiongrad
(dy)
dnnlib/tflib/ops/upfirdn_2d.py:138
Functiongrad
(dy)
dnnlib/tflib/ops/fused_bias_act.py:180
Functiongrad2
(d_dx, d_db)
dnnlib/tflib/ops/fused_bias_act.py:183
Functiongrad_wrap
(dy)
dnnlib/tflib/ops/fused_bias_act.py:193
Functionhdf5_from_images
(hdf5_filename, image_dir, compress)
dataset_tool.py:715
Methodimages_float
(self)
projector.py:196
Methodimages_uint8
(self)
projector.py:200
Functionimport_handler
Function decorator for declaring custom import handlers.
dnnlib/tflib/network.py:36
Functioninfo
(tfrecord_dir)
dataset_tool.py:298
Functioninit_tf
Initialize TensorFlow session using good default settings.
dnnlib/tflib/tfutil.py:117
Functioninit_uninitialized_vars
Initialize all tf.Variables that have not already been initialized. Equivalent to the following, but more efficient and does not bloat the tf gra
dnnlib/tflib/tfutil.py:174
Methodinput_names
Name string for each input.
dnnlib/tflib/network.py:250
Methodinput_shape
Short-hand for input_shapes[0].
dnnlib/tflib/network.py:230
Methodinput_shapes
List of input tensor shapes, including minibatch dimension.
dnnlib/tflib/network.py:216
Methodinput_templates
Input placeholders in the template graph.
dnnlib/tflib/network.py:264
Functionis_pickleable
(obj: Any)
dnnlib/util.py:207
Functionlerp
Linear interpolation.
dnnlib/tflib/tfutil.py:71
Functionlerp_clip
Linear interpolation with clip.
dnnlib/tflib/tfutil.py:77
Functionlist_dir_recursively_with_ignore
List all files recursively in a given directory while ignoring given file and directory names. Returns list of tuples containing both absolute and
dnnlib/util.py:312
Functionload_dataset
(path=None, resolution=None, max_images=None, max_label_size=0, mirror_augment=False, repeat=True, shuffle=Tru
training/dataset.py:225
← previousnext →201–300 of 336, ranked by callers