MCPcopy Create free account

hub / github.com/apple/ml-facelit / functions

Functions451 in github.com/apple/ml-facelit

↓ 2 callersFunctionslerp
(a, b, t)
facelit/metrics/perceptual_path_length.py:24
↓ 2 callersFunctionsuppress_tracer_warnings
()
facelit/torch_utils/misc.py:73
↓ 1 callersFunctionFOV_to_intrinsics
Creates a 3x3 camera intrinsics matrix from the camera field of view, specified in degrees. Note the intrinsics are returned as normalized by
facelit/camera_utils.py:142
↓ 1 callersFunctionSH_basis
https://github.com/zhhoper/DPR/blob/master/utils/utils_SH.py get SH basis based on normal normal is a Nx3 matrix retu
facelit/light_utils.py:94
↓ 1 callersMethod__getstate__
(self)
facelit/training/dataset.py:87
↓ 1 callersMethod__init__
(self, name, # Name of the dataset. raw_shape, # Shape of the r
facelit/training/dataset.py:29
↓ 1 callersMethod__reduce__
(self)
facelit/torch_utils/persistence.py:120
↓ 1 callersFunction_bias_act_cuda
Fast CUDA implementation of `bias_act()` using custom ops.
facelit/torch_utils/ops/bias_act.py:128
↓ 1 callersFunction_bias_act_ref
Slow reference implementation of `bias_act()` using standard TensorFlow ops.
facelit/torch_utils/ops/bias_act.py:93
↓ 1 callersFunction_check_pickleable
r"""Check that the given object is pickleable, raising an exception if it is not. This function is expected to be considerably more efficient
facelit/torch_utils/persistence.py:233
↓ 1 callersFunction_filtered_lrelu_ref
Slow and memory-inefficient reference implementation of `filtered_lrelu()` using existing `upfirdn2n()` and `bias_act()` ops.
facelit/torch_utils/ops/filtered_lrelu.py:123
↓ 1 callersFunction_find_compiler_bindir
()
facelit/torch_utils/custom_ops.py:31
↓ 1 callersFunction_get_mangled_gpu_name
()
facelit/torch_utils/custom_ops.py:46
↓ 1 callersFunction_init
()
facelit/torch_utils/ops/filtered_lrelu.py:25
↓ 1 callersFunction_init
()
facelit/torch_utils/ops/bias_act.py:40
↓ 1 callersFunction_init
()
facelit/torch_utils/ops/upfirdn2d.py:25
↓ 1 callersMethod_load_deca_labels
(self, use_light=True)
facelit/training/dataset.py:257
↓ 1 callersMethod_load_raw_image
(self, raw_idx)
facelit/training/dataset.py:81
↓ 1 callersMethod_load_raw_image
(self, raw_idx)
facelit/training/dataset.py:228
↓ 1 callersMethod_load_raw_labels
(self)
facelit/training/dataset.py:84
↓ 1 callersFunction_module_to_src
r"""Query the source code of a given Python module.
facelit/torch_utils/persistence.py:208
↓ 1 callersFunction_should_use_custom_op
()
facelit/torch_utils/ops/grid_sample_gradfix.py:35
↓ 1 callersFunction_src_to_module
r"""Get or create a Python module for the given source code.
facelit/torch_utils/persistence.py:218
↓ 1 callersFunction_sync
r"""Synchronize the global cumulative counters across devices and processes. Called internally by `Collector.update()`.
facelit/torch_utils/training_stats.py:236
↓ 1 callersFunction_upfirdn2d_ref
Slow reference implementation of `upfirdn2d()` using standard PyTorch ops.
facelit/torch_utils/ops/upfirdn2d.py:169
↓ 1 callersMethodaccumulate_gradients
(self, phase, real_img, real_c, gen_z, gen_c, gain, cur_nimg)
facelit/training/loss.py:23
↓ 1 callersFunctionapply_affine_transformation
(x, mat, up=4, **filter_kwargs)
facelit/metrics/equivariance.py:137
↓ 1 callersFunctionapply_fractional_pseudo_rotation
(x, angle, a=3, **filter_kwargs)
facelit/metrics/equivariance.py:181
↓ 1 callersFunctionapply_fractional_rotation
(x, angle, a=3, **filter_kwargs)
facelit/metrics/equivariance.py:171
↓ 1 callersFunctionapply_fractional_translation
(x, tx, ty, a=3)
facelit/metrics/equivariance.py:65
↓ 1 callersFunctionapply_integer_translation
(x, tx, ty)
facelit/metrics/equivariance.py:46
↓ 1 callersMethodas_dict
r"""Returns the averages accumulated between the last two calls to `update()` as an `dnnlib.EasyDict`. The contents are as follows:
facelit/torch_utils/training_stats.py:214
↓ 1 callersFunctionbatch_rodrigues
Convert axis-angle representation to rotation matrix. Args: theta: size = [B, 3] Returns: Rotation matrix corresponding to the
facelit/geometry_utils.py:26
↓ 1 callersFunctioncalc_deca_consistency_metrics
(eval_dir)
facelit/calc_deca_consistency.py:14
↓ 1 callersFunctioncalc_face_consistency_metrics
(img_dir)
facelit/calc_face_consistency.py:11
↓ 1 callersFunctioncalc_metrics
Calculate quality metrics for previous training run or pretrained network pickle. Examples: \b # Previous training run: look up options
facelit/calc_metrics.py:114
↓ 1 callersFunctioncall_func_by_name
Finds the python object with the given name and calls it as a function.
facelit/dnnlib/util.py:295
↓ 1 callersFunctioncompute_depth_err
(alinged_depth_f, pred_depth_f, pred_mask_f)
facelit/calc_geometry_metrics.py:33
↓ 1 callersFunctionconvert_dataset
Convert an image dataset into a dataset archive usable with StyleGAN2 ADA PyTorch. The input dataset format is guessed from the --source argument
facelit/dataset_tool.py:327
↓ 1 callersFunctionconvert_network_pickle
Convert legacy network pickle into the native PyTorch format. The tool is able to load the main network configurations exported using the TensorF
facelit/legacy.py:299
↓ 1 callersFunctionconvert_tf_discriminator
(tf_D)
facelit/legacy.py:210
↓ 1 callersFunctioncreate_samples
(N=256, voxel_origin=[0, 0, 0], cube_length=2.0)
facelit/gen_videos.py:51
↓ 1 callersFunctioncreate_samples
(N=256, voxel_origin=[0, 0, 0], cube_length=2.0)
facelit/gen_samples.py:81
↓ 1 callersFunctionestimate_translation_np
Find camera translation that brings 3D joints S closest to 2D the corresponding joints_2d. Input: S: (25, 3) 3D joint locations jo
facelit/geometry_utils.py:111
↓ 1 callersFunctiongenerate_images
Render a latent vector interpolation video. Examples: \b # Render a 4x2 grid of interpolations for seeds 0 through 31. python gen_vi
facelit/gen_videos.py:311
↓ 1 callersFunctiongenerate_images
Generate images using pretrained network pickle. Examples: \b # Generate an image using pre-trained FFHQ model. python gen_samples.p
facelit/gen_samples.py:124
↓ 1 callersFunctiongenerate_planes
Defines planes by the three vectors that form the "axes" of the plane. Should work with arbitrary number of planes and planes of arbitrar
facelit/training/volumetric_rendering/renderer.py:61
↓ 1 callersMethodget_details
(self, idx)
facelit/training/dataset.py:117
↓ 1 callersFunctionget_feature_detector_name
(url)
facelit/metrics/metric_utils.py:41
↓ 1 callersFunctionget_file_ext
(fname)
facelit/prepare_face_landmarks.py:33
↓ 1 callersMethodget_label_std
(self)
facelit/training/dataset.py:124
↓ 1 callersFunctionget_normals
https://github.com/zhhoper/DPR/blob/master/testNetwork_demo_512.py
facelit/light_utils.py:61
↓ 1 callersFunctionget_obj_by_name
Finds the python object with the given name.
facelit/dnnlib/util.py:289
↓ 1 callersFunctionget_shading
https://github.com/zhhoper/DPR/blob/master/utils/utils_SH.py get shading based on normals and SH normal is Nx3 matrix
facelit/light_utils.py:80
↓ 1 callersFunctioninit_dataset_kwargs
(data, pose_cond, use_deca, use_light)
facelit/train.py:107
↓ 1 callersMethodis_full
(self)
facelit/metrics/metric_utils.py:93
↓ 1 callersFunctionis_top_level_function
Determine whether the given object is a top-level function, i.e., defined at module scope using 'def'.
facelit/dnnlib/util.py:314
↓ 1 callersFunctionis_url
Determine whether the given object is a valid URL string.
facelit/dnnlib/util.py:380
↓ 1 callersFunctioniterate_random_labels
(opts, batch_size)
facelit/metrics/metric_utils.py:59
↓ 1 callersFunctionlaunch_training
(c, desc, outdir, dry_run)
facelit/train.py:56
↓ 1 callersFunctionlayout_grid
(img, grid_w=None, grid_h=1, float_to_uint8=True, chw_to_hwc=True, to_numpy=True)
facelit/gen_videos.py:35
↓ 1 callersFunctionload_network_pkl
(f, force_fp16=False)
facelit/legacy.py:24
↓ 1 callersMethodload_samples_deca
(self)
facelit/light_utils.py:19
↓ 1 callersMethodload_samples_deca
(self)
facelit/deca_utils.py:20
↓ 1 callersFunctionmain
Train a GAN using the techniques described in the paper "Alias-Free Generative Adversarial Networks". Examples: \b # Train StyleGAN3
facelit/train.py:206
↓ 1 callersFunctionmain
(args)
facelit/prepare_face_landmarks.py:37
↓ 1 callersFunctionmake_cache_dir_path
(*paths: str)
facelit/dnnlib/util.py:126
↓ 1 callersFunctionmake_transform
( transform: Optional[str], output_width: Optional[int], output_height: Optional[int] )
facelit/dataset_tool.py:217
↓ 1 callersFunctionmodulated_conv2d
( x, # Input tensor: [batch_size, in_channels, in_height, in_width] w,
facelit/training/networks_stylegan3.py:27
↓ 1 callersFunctionnan_to_num
(input, nan=0.0, posinf=None, neginf=None, *, out=None)
facelit/torch_utils/misc.py:51
↓ 1 callersMethodnum
r"""Returns the number of scalars that were accumulated for the given statistic between the last two calls to `update()`, or zero if n
facelit/torch_utils/training_stats.py:182
↓ 1 callersFunctionopen_cifar10
(tarball: str, *, max_images: Optional[int])
facelit/dataset_tool.py:155
↓ 1 callersFunctionopen_dataset
(source, *, max_images: Optional[int])
facelit/dataset_tool.py:268
↓ 1 callersFunctionopen_dest
(dest: str)
facelit/dataset_tool.py:288
↓ 1 callersFunctionopen_image_folder
(source_dir, *, max_images: Optional[int])
facelit/dataset_tool.py:70
↓ 1 callersFunctionopen_image_zip
(source, *, max_images: Optional[int])
facelit/dataset_tool.py:98
↓ 1 callersFunctionopen_lmdb
(lmdb_dir: str, *, max_images: Optional[int])
facelit/dataset_tool.py:127
↓ 1 callersFunctionopen_mnist
(images_gz: str, *, max_images: Optional[int])
facelit/dataset_tool.py:187
↓ 1 callersFunctionpersistent_class
r"""Class decorator that extends a given class to save its source code when pickled. Example: from torch_utils import persistence
facelit/torch_utils/persistence.py:37
↓ 1 callersFunctionproject_onto_planes
Does a projection of a 3D point onto a batch of 2D planes, returning 2D plane coordinates. Takes plane axes of shape n_planes, 3, 3
facelit/training/volumetric_rendering/renderer.py:77
↓ 1 callersFunctionquat_to_rotmat
Convert quaternion coefficients to rotation matrix. Args: quat: size = [B, 4] 4 <===>(w, x, y, z) Returns: Rotation matrix cor
facelit/geometry_utils.py:42
↓ 1 callersFunctionrecurse
(prefix, tf_net)
facelit/legacy.py:78
↓ 1 callersFunctionrecurse
(obj)
facelit/torch_utils/persistence.py:238
↓ 1 callersFunctionreport
r"""Broadcasts the given set of scalars to all interested instances of `Collector`, across device and process boundaries. This function is ex
facelit/torch_utils/training_stats.py:58
↓ 1 callersFunctionrotate2d
(theta, **kwargs)
facelit/training/augment.py:90
↓ 1 callersFunctionrotate3d
(v, theta, **kwargs)
facelit/training/augment.py:97
↓ 1 callersMethodrun_forward
(self, colors, densities, depths, rendering_options)
facelit/training/volumetric_rendering/ray_marcher.py:25
↓ 1 callersFunctionsample_from_planes
(plane_axes, plane_features, coordinates, mode='bilinear', padding_mode='zeros', box_warp=None)
facelit/training/volumetric_rendering/renderer.py:93
↓ 1 callersMethodsample_importance
Return depths of importance sampled points along rays. See NeRF importance sampling for more.
facelit/training/volumetric_rendering/renderer.py:279
↓ 1 callersMethodsample_pdf
Sample @N_importance samples from @bins with distribution defined by @weights. Inputs: bins: (N_rays, N_samples_+1) where
facelit/training/volumetric_rendering/renderer.py:299
↓ 1 callersFunctionscale3d
(sx, sy, sz, **kwargs)
facelit/training/augment.py:82
↓ 1 callersMethodset_num_features
(self, num_features)
facelit/metrics/metric_utils.py:84
↓ 1 callersFunctionsetup_snapshot_image_grid
(training_set, random_seed=0)
facelit/training/training_loop.py:35
↓ 1 callersFunctionshtools_matrix2vec
for the sh matrix created by sh tools, we create the vector of the sh
facelit/light_utils.py:142
↓ 1 callersFunctionshtools_sh2matrix
convert vector of sh to matrix
facelit/light_utils.py:159
↓ 1 callersFunctionsubprocess_fn
(rank, c, temp_dir)
facelit/train.py:32
↓ 1 callersFunctionsubprocess_fn
(rank, args, temp_dir)
facelit/calc_metrics.py:31
↓ 1 callersMethodsynthesis_with_light
(self, ws, c, diffuse, specular, neural_rendering_resolution=None, update_emas=False, cache_backbone=False, us
facelit/training/triplane.py:126
↓ 1 callersFunctiontranslate3d
(tx, ty, tz, **kwargs)
facelit/training/augment.py:67
↓ 1 callersMethodunify_samples
(self, depths1, colors1, densities1, depths2, colors2, densities2)
facelit/training/volumetric_rendering/renderer.py:242
← previousnext →101–200 of 451, ranked by callers