MCPcopy Create free account

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

Functions451 in github.com/apple/ml-facelit

↓ 47 callersFunctionkwarg
(tf_name, default=None, none=None)
facelit/legacy.py:116
↓ 39 callersMethodmean
r"""Returns the mean of the scalars that were accumulated for the given statistic between the last two calls to `update()`, or NaN if
facelit/torch_utils/training_stats.py:190
↓ 33 callersMethodappend
(self, x)
facelit/metrics/metric_utils.py:96
↓ 26 callersMethodmapping
(self, z, c, truncation_psi=1, truncation_cutoff=None, update_emas=False)
facelit/training/triplane.py:56
↓ 19 callersMethodload
(pkl_file)
facelit/metrics/metric_utils.py:144
↓ 19 callersMethodsynthesis
(self, ws, c, neural_rendering_resolution=None, update_emas=False, cache_backbone=False, use_cached_backbone=F
facelit/training/triplane.py:61
↓ 17 callersMethodupdate
(self, cur_items)
facelit/metrics/metric_utils.py:169
↓ 13 callersMethodsave
(self, pkl_file)
facelit/metrics/metric_utils.py:139
↓ 11 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
facelit/training/networks_stylegan2.py:530
↓ 11 callersMethodsample
(self)
facelit/deca_utils.py:36
↓ 9 callersFunctionsinc
(x)
facelit/metrics/equivariance.py:24
↓ 9 callersMethodupdate
r"""Copies current values of the internal counters to the user-visible state and resets them for the next round. If `keep_previous=Tr
facelit/torch_utils/training_stats.py:149
↓ 8 callersMethodbackward
(ctx, dy)
facelit/torch_utils/ops/bias_act.py:160
↓ 8 callersFunctionerror
(msg)
facelit/dataset_tool.py:33
↓ 8 callersMethodstd
r"""Returns the standard deviation of the scalars that were accumulated for the given statistic between the last two calls to `update(
facelit/torch_utils/training_stats.py:200
↓ 8 callersMethodsub
(self, tag=None, num_items=None, flush_interval=1000, rel_lo=0, rel_hi=1)
facelit/metrics/metric_utils.py:184
↓ 7 callersFunctionmatrix
(*rows, device=None)
facelit/training/augment.py:50
↓ 7 callersMethodsample_mixed
(self, coordinates, directions, ws, truncation_psi=1, truncation_cutoff=None, update_emas=False, **synthesis_k
facelit/training/triplane.py:115
↓ 7 callersMethodwrite
Write text to stdout (and a file) and optionally flush.
facelit/dnnlib/util.py:80
↓ 6 callersFunction_conv2d_wrapper
Wrapper for the underlying `conv2d()` and `conv_transpose2d()` implementations.
facelit/torch_utils/ops/conv2d_resample.py:31
↓ 6 callersFunction_parse_padding
(padding)
facelit/torch_utils/ops/upfirdn2d.py:46
↓ 6 callersFunction_parse_scaling
(scaling)
facelit/torch_utils/ops/upfirdn2d.py:37
↓ 6 callersFunctionrotate_SH_coeffs
(sh, angles, dj=None)
facelit/light_utils.py:173
↓ 5 callersMethod__init__
(self, z_dim, # Input latent (Z) dimensionality. c_dim,
facelit/training/networks_stylegan3.py:493
↓ 5 callersMethod__init__
(self, in_channels, # Number of input channels, 0 = first block. ou
facelit/training/superresolution.py:159
↓ 5 callersFunction_get_filter_size
(f)
facelit/torch_utils/ops/upfirdn2d.py:57
↓ 5 callersFunctionangle_in_a_circle
(param, axis='z')
facelit/light_utils.py:198
↓ 5 callersFunctioniterate_images
()
facelit/dataset_tool.py:86
↓ 5 callersFunctionmaybe_min
(a: int, b: Optional[int])
facelit/dataset_tool.py:52
↓ 5 callersFunctionscale2d_inv
(sx, sy, **kwargs)
facelit/training/augment.py:110
↓ 4 callersFunction_conv2d_gradfix
(transpose, weight_shape, stride, padding, output_padding, dilation, groups)
facelit/torch_utils/ops/conv2d_gradfix.py:68
↓ 4 callersMethod_get_raw_labels
(self)
facelit/training/dataset.py:59
↓ 4 callersFunction_tuple_of_ints
(xs, ndim)
facelit/torch_utils/ops/conv2d_gradfix.py:57
↓ 4 callersFunctionfiltered_resizing
(image_orig_tensor, size, f, filter_mode='antialiased')
facelit/training/dual_discriminator.py:86
↓ 4 callersMethodflush
Flush written text to both stdout and a file, if open.
facelit/dnnlib/util.py:95
↓ 4 callersMethodget_all
(self)
facelit/metrics/metric_utils.py:125
↓ 4 callersFunctionhas_same_layout
facelit/torch_utils/ops/bias_act.cpp:20
↓ 4 callersFunctionlanczos_window
(x, a)
facelit/metrics/equivariance.py:29
↓ 4 callersMethodload_example_light
(self)
facelit/light_utils.py:43
↓ 4 callersMethodrun_model
(self, planes, decoder, sample_coordinates, sample_directions, options)
facelit/training/volumetric_rendering/renderer.py:227
↓ 4 callersFunctionsave_image_grid
(img, fname, drange, grid_size)
facelit/training/training_loop.py:75
↓ 3 callersMethod__init__
(self, n_features, options)
facelit/training/triplane.py:176
↓ 3 callersMethod__init__
(self, c_dim, # Conditioning label (C) dimensionality. img_resolution
facelit/training/dual_discriminator.py:108
↓ 3 callersMethod_file_ext
(fname)
facelit/training/dataset.py:202
↓ 3 callersMethod_get_delta
r"""Returns the raw moments that were accumulated for the given statistic between the last two calls to `update()`, or zero if no scal
facelit/torch_utils/training_stats.py:172
↓ 3 callersFunction_unbroadcast
(x, shape)
facelit/torch_utils/ops/fma.py:51
↓ 3 callersFunctionbatch_reflection
incident: B x N x 3 normal: B x N x 3 w_r = 2(w0.n)n - w0
facelit/geometry_utils.py:207
↓ 3 callersFunctionconvert_sdf_samples_to_ply
Convert sdf samples to .ply :param pytorch_3d_sdf_tensor: a torch.FloatTensor of shape (n,n,n) :voxel_grid_origin: a list of three floats
facelit/shape_utils.py:40
↓ 3 callersFunctioncreate_cam2world_matrix
Takes in the direction the camera is pointing and the camera origin and returns a cam2world matrix. Works on batches of forward_vectors, orig
facelit/camera_utils.py:120
↓ 3 callersFunctionfile_ext
(name: Union[str, Path])
facelit/dataset_tool.py:59
↓ 3 callersFunctionget_extrinsics_from_axis_angle_and_cam
(axis_angle, cam)
facelit/geometry_utils.py:10
↓ 3 callersMethodget_label
(self, idx)
facelit/training/dataset.py:109
↓ 3 callersFunctionget_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
facelit/dnnlib/util.py:279
↓ 3 callersFunctionnamed_params_and_buffers
(module)
facelit/torch_utils/misc.py:153
↓ 3 callersFunctionpaste_light_on_img_tensor
sphere_size: int, denoting SxS sized half sphere light_coeff: 9x3 tensor of sh coefficient img: BxCxHxW batched images
facelit/light_utils.py:183
↓ 3 callersFunctionrender_half_sphere
sh: np.array (9x3) https://github.com/zhhoper/DPR/blob/master/testNetwork_demo_512.py
facelit/light_utils.py:124
↓ 3 callersFunctionrotate2d_inv
(theta, **kwargs)
facelit/training/augment.py:113
↓ 3 callersFunctionrotation_matrix
(angle)
facelit/metrics/equivariance.py:33
↓ 3 callersMethodrun_D
(self, img, c, blur_sigma=0, blur_sigma_raw=0, update_emas=False)
facelit/training/loss.py:74
↓ 3 callersFunctionscale2d
(sx, sy, **kwargs)
facelit/training/augment.py:75
↓ 3 callersFunctiontranslate2d
(tx, ty, **kwargs)
facelit/training/augment.py:60
↓ 3 callersFunctiontranslate2d_inv
(tx, ty, **kwargs)
facelit/training/augment.py:107
↓ 3 callersFunctionupfirdn2d
r"""Pad, upsample, filter, and downsample a batch of 2D images. Performs the following sequence of operations for each channel: 1. Upsample
facelit/torch_utils/ops/upfirdn2d.py:120
↓ 2 callersFunction_collect_tf_params
(tf_net)
facelit/legacy.py:75
↓ 2 callersFunction_filtered_lrelu_cuda
Fast CUDA implementation of `filtered_lrelu()` using custom ops.
facelit/torch_utils/ops/filtered_lrelu.py:161
↓ 2 callersFunction_get_filter_size
(f)
facelit/torch_utils/ops/filtered_lrelu.py:37
↓ 2 callersFunction_get_weight_shape
(w)
facelit/torch_utils/ops/conv2d_resample.py:23
↓ 2 callersMethod_get_zipfile
(self)
facelit/training/dataset.py:205
↓ 2 callersMethod_open_file
(self, fname)
facelit/training/dataset.py:211
↓ 2 callersFunction_parse_padding
(padding)
facelit/torch_utils/ops/filtered_lrelu.py:44
↓ 2 callersFunction_populate_module_params
(module, *patterns)
facelit/legacy.py:88
↓ 2 callersFunction_should_use_custom_op
(input)
facelit/torch_utils/ops/conv2d_gradfix.py:49
↓ 2 callersFunction_upfirdn2d_cuda
Fast CUDA implementation of `upfirdn2d()` using custom ops.
facelit/torch_utils/ops/upfirdn2d.py:219
↓ 2 callersMethodappend_torch
(self, x, num_gpus=1, rank=0)
facelit/metrics/metric_utils.py:113
↓ 2 callersFunctioncalc_output_padding
(input_shape, output_shape)
facelit/torch_utils/ops/conv2d_gradfix.py:95
↓ 2 callersMethodclose
(self)
facelit/training/dataset.py:78
↓ 2 callersMethodclose
Flush, close possible files, and remove stdout/stderr mirroring.
facelit/dnnlib/util.py:102
↓ 2 callersFunctioncompute_diffuse_shading
albedo: b x N x F normal: b x N x 3 sh_light: b x 9 x 3 constant_factor: 9
facelit/training/volumetric_rendering/renderer.py:24
↓ 2 callersFunctioncompute_distances
(row_features, col_features, num_gpus, rank, col_batch_size)
facelit/metrics/precision_recall.py:21
↓ 2 callersFunctioncompute_normals
Args: inputs (B, N, d): outputs (B, N): Returns:
facelit/geometry_utils.py:177
↓ 2 callersFunctioncompute_specular_shading
albedo: b x N x F reflect_ray: b x N x 3 sh_light: b x 9 x 3 k_s : b x N x 1 constant_factor: 9
facelit/training/volumetric_rendering/renderer.py:42
↓ 2 callersFunctionconstruct_affine_bandlimit_filter
(mat, a=3, amax=16, aflt=64, up=4, cutoff_in=1, cutoff_out=1)
facelit/metrics/equivariance.py:104
↓ 2 callersFunctionconvert_mrc
(input_filename, output_filename, isosurface_level=1)
facelit/shape_utils.py:109
↓ 2 callersFunctionconvert_tf_generator
(tf_G)
facelit/legacy.py:109
↓ 2 callersMethoddesign_lowpass_filter
(numtaps, cutoff, width, fs, radial=False)
facelit/training/networks_stylegan3.py:366
↓ 2 callersFunctiongen_interp_video
(G, mp4: str, seeds, shuffle_seed=None, w_frames=60*4, kind='cubic', grid_dims=(1,1), num_keyframes=None, wrap
facelit/gen_videos.py:82
↓ 2 callersMethodget_all_torch
(self)
facelit/metrics/metric_utils.py:129
↓ 2 callersFunctionget_feature_detector
(url, device=torch.device('cpu'), num_gpus=1, rank=0, verbose=False)
facelit/metrics/metric_utils.py:44
↓ 2 callersMethodget_mean_cov
(self)
facelit/metrics/metric_utils.py:132
↓ 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
facelit/dnnlib/util.py:238
↓ 2 callersFunctionis_image_ext
(fname: Union[str, Path])
facelit/dataset_tool.py:64
↓ 2 callersFunctionis_persistent
r"""Test whether the given object or class is persistent, i.e., whether it will save its source code when pickled.
facelit/torch_utils/persistence.py:136
↓ 2 callersFunctionis_valid_metric
(metric)
facelit/metrics/metric_main.py:36
↓ 2 callersMethodload_deca_center_light
(self)
facelit/light_utils.py:31
↓ 2 callersFunctionmodulated_conv2d
( x, # Input tensor of shape [batch_size, in_channels, in_height, in_width].
facelit/training/networks_stylegan2.py:34
↓ 2 callersMethodnames
r"""Returns the names of all statistics broadcasted so far that match the regular expression specified at construction time.
facelit/torch_utils/training_stats.py:143
↓ 2 callersFunctionnormalize_2nd_moment
(x, dim=1, eps=1e-8)
facelit/training/networks_stylegan2.py:28
↓ 2 callersFunctionnormalize_vec
(vec)
facelit/calc_geometry_metrics.py:26
↓ 2 callersMethodrun_G
(self, z, c, swapping_prob, neural_rendering_resolution, update_emas=False)
facelit/training/loss.py:58
↓ 2 callersMethodsample_stratified
Return depths of approximately uniformly spaced samples along rays.
facelit/training/volumetric_rendering/renderer.py:254
next →1–100 of 451, ranked by callers