MCPcopy Create free account

hub / github.com/ant-research/TensorialGaussianAvatar / functions

Functions250 in github.com/ant-research/TensorialGaussianAvatar

↓ 22 callersMethoditems
(self)
flame_model/flame.py:577
↓ 20 callersMethodload
(self)
utils/viewer_utils.py:117
↓ 13 callersFunctionread_next_bytes
Read and unpack the next bytes from a binary file. :param fid: :param num_bytes: Sum of combination of {2, 4, 8}, e.g. 2, 6, 16, 30, etc.
scene/colmap_loader.py:72
↓ 10 callersMethodwrite
(self, x)
utils/general_utils.py:118
↓ 8 callersFunctionreadCamerasFromTransforms
(path, transformsfile, white_background, extension=".png")
scene/dataset_readers.py:189
↓ 7 callersMethodpan
(self, dx=0, dy=0, dz=0)
utils/viewer_utils.py:190
↓ 6 callersFunctionaxis_angle_to_quaternion
(axis_angle: torch.Tensor)
scene/flame_gaussian_model.py:29
↓ 6 callersMethodextract
(self, args)
arguments/__init__.py:40
↓ 6 callersMethodselect_mesh_by_timestep
(self, timestep)
scene/gaussian_model.py:227
↓ 6 callersFunctionto_np
(array, dtype=np.float32)
flame_model/flame.py:45
↓ 6 callersFunctionto_tensor
(array, dtype=torch.float32)
flame_model/flame.py:40
↓ 5 callersFunctionfocal2fov
(focal, pixels)
utils/graphics_utils.py:76
↓ 5 callersMethodsave
(self, iteration)
scene/__init__.py:155
↓ 4 callersMethod__init__
(self)
lpipsPyTorch/modules/networks.py:89
↓ 4 callersFunction_make_tensor
Return a 2D tensor with the specified cols and dtype filled with data, even when data is empty.
utils/pytorch3d_load_obj.py:74
↓ 4 callersFunctiondot
(x: torch.Tensor, y: torch.Tensor)
utils/graphics_utils.py:90
↓ 4 callersMethodgetTrainCameras
(self, scale=1.0)
scene/__init__.py:159
↓ 4 callersFunctionreadMeshesFromTransforms
(path, transformsfile)
scene/dataset_readers.py:284
↓ 4 callersFunctionrender_set
(dataset : ModelParams, name, iteration, views, gaussians, pipeline, background, render_mesh)
render.py:52
↓ 4 callersFunctionsafe_normalize
(x: torch.Tensor, eps: float =1e-20)
utils/graphics_utils.py:99
↓ 3 callersMethod__init__
(self, **kwargs)
flame_model/flame.py:52
↓ 3 callersMethod__init__
(self, parser: ArgumentParser, name : str, fill_none = False)
arguments/__init__.py:20
↓ 3 callersFunction_format_faces_indices
Format indices and check for invalid values. Indices can refer to values in one of the face properties: vertices, textures or normals. Se
utils/pytorch3d_load_obj.py:108
↓ 3 callersFunctioncameraList_from_camInfos
(cam_infos, resolution_scale, args)
utils/camera_utils.py:51
↓ 3 callersFunctioncompute_face_orientation
(verts, faces, return_scale=False)
utils/graphics_utils.py:116
↓ 3 callersFunctionfov2focal
(fov, pixels)
utils/graphics_utils.py:73
↓ 3 callersFunctiongetNerfppNorm
(cam_info)
scene/dataset_readers.py:54
↓ 3 callersFunctionsample_feature_line
(bs_feat_line, jaw_feat_line, pts)
utils/feature_line.py:8
↓ 3 callersMethodset_requires_grad
(self, state: bool)
lpipsPyTorch/modules/networks.py:46
↓ 3 callersMethodupdate
Update the faces properties when vertex masks are changed
flame_model/flame.py:609
↓ 2 callersFunctionSH2RGB
(sh)
utils/sh_utils.py:117
↓ 2 callersFunctionbuild_rotation
(r)
utils/general_utils.py:78
↓ 2 callersMethodcat_tensors_to_optimizer
(self, tensors_dict)
scene/gaussian_model.py:475
↓ 2 callersMethodconstruct_vid_table
(self)
flame_model/flame.py:816
↓ 2 callersFunctioncustom_distance
(rest_points, sampled_points, batch_size=64)
expr_analyze.py:28
↓ 2 callersMethoddensification_postfix
(self, new_xyz, new_features_dc, new_features_rest, new_opacities, new_scaling, new_rotation)
scene/gaussian_model.py:518
↓ 2 callersMethoddensification_postfix_triplane
(self, new_xyz, new_opacities, new_scaling, new_rotation)
scene/gaussian_model.py:501
↓ 2 callersFunctiondot
(x: torch.Tensor, y: torch.Tensor)
utils/vector_ops.py:13
↓ 2 callersFunctionfetchPly
(path)
scene/dataset_readers.py:117
↓ 2 callersMethodgetTestCameras
(self, scale=1.0)
scene/__init__.py:165
↓ 2 callersMethodgetValCameras
(self, scale=1.0)
scene/__init__.py:162
↓ 2 callersFunctiongetWorld2View2
(R, t, translate=np.array([.0, .0, .0]), scale=1.0)
utils/graphics_utils.py:38
↓ 2 callersMethodget_opacity
(self)
scene/gaussian_model.py:210
↓ 2 callersMethodget_vid_by_region
Get vertex indicies by regions
flame_model/flame.py:869
↓ 2 callersFunctioninverse_sigmoid
(x)
utils/general_utils.py:18
↓ 2 callersFunctionl1_loss
(network_output, gt)
utils/loss_utils.py:17
↓ 2 callersFunctionlength
(x: torch.Tensor, eps: float =1e-20)
utils/graphics_utils.py:96
↓ 2 callersMethodprocess_face_clusters
Construct a lookup table from face id to cluster id. cluster #0: background cluster #1: foreground c
flame_model/flame.py:840
↓ 2 callersMethodprocess_face_mask
(self, faces)
flame_model/flame.py:822
↓ 2 callersMethodprocess_vt_mask
(self, faces, faces_t)
flame_model/flame.py:858
↓ 2 callersMethodprune_points
(self, mask)
scene/gaussian_model.py:445
↓ 2 callersFunctionqvec2rotmat
(qvec)
scene/colmap_loader.py:43
↓ 2 callersFunctionrender
Render the scene. Background tensor (bg_color) must be on GPU!
gaussian_renderer/__init__.py:19
↓ 2 callersFunctionsafe_state
(silent)
utils/general_utils.py:112
↓ 2 callersFunctionssim
(img1, img2, window_size=11, size_average=True)
utils/loss_utils.py:33
↓ 2 callersFunctionstorePly
(path, xyz, rgb)
scene/dataset_readers.py:125
↓ 2 callersMethodtraining_setup
(self, training_args)
scene/gaussian_model.py:274
↓ 2 callersMethodupdate_offset
(self)
scene/flame_gaussian_model.py:316
↓ 1 callersFunctionPILtoTorch
(pil_image, resolution)
utils/general_utils.py:21
↓ 1 callersFunctionRGB2SH
(rgb)
utils/sh_utils.py:114
↓ 1 callersMethod__init__
(self, **kwargs)
scene/flame_gaussian_model.py:25
↓ 1 callersMethod__init__
(self, mlptype, dim_in, dim_out, dim_hidden, num_layers)
utils/triplane_utils.py:78
↓ 1 callersFunction_check_faces_indices
( faces_indices: torch.Tensor, max_index: int, pad_value: Optional[int] = None )
utils/pytorch3d_load_obj.py:87
↓ 1 callersFunction_load_materials
Load materials and optionally textures from the specified path. Args: material_names: a list of the material names found in the .obj
utils/pytorch3d_load_obj.py:463
↓ 1 callersFunction_load_obj
Load a mesh from a file-like object. See load_obj function more details. Any material files associated with the obj are expected to be in the
utils/pytorch3d_load_obj.py:509
↓ 1 callersFunction_open_file
(f, path_manager: PathManager, mode: str = "r")
utils/pytorch3d_load_obj.py:63
↓ 1 callersFunction_parse_face
( line, tokens, material_idx, faces_verts_idx, faces_normals_idx, faces_textures_idx,
utils/pytorch3d_load_obj.py:309
↓ 1 callersFunction_parse_obj
Load a mesh from a file-like object. See load_obj function for more details about the return values.
utils/pytorch3d_load_obj.py:380
↓ 1 callersMethod_prune_optimizer
(self, mask)
scene/gaussian_model.py:423
↓ 1 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average=True)
utils/loss_utils.py:43
↓ 1 callersMethodadd_densification_stats
(self, viewspace_point_tensor, update_filter)
scene/gaussian_model.py:619
↓ 1 callersMethodadd_teeth
(self)
flame_model/flame.py:186
↓ 1 callersFunctionaxis_angle_to_quaternion
Convert rotations given as axis/angle to quaternions. Args: axis_angle: Rotations given as a vector in axis angle form,
preprocess_jaw.py:35
↓ 1 callersFunctionbatch_rigid_transform
Applies a batch of rigid transformations to the joints Parameters ---------- rot_mats : torch.tensor BxNx3x3 Tensor of rotat
flame_model/lbs.py:254
↓ 1 callersFunctionbatch_rodrigues
Calculates the rotation matrices for a batch of rotation vectors Parameters ---------- rot_vecs: torch.tensor Nx3 array of N axis-
flame_model/lbs.py:25
↓ 1 callersFunctionblend_shapes
Calculates the per vertex displacement due to the blend shapes Parameters ---------- betas : torch.tensor Bx(num_betas) Blend sh
flame_model/lbs.py:218
↓ 1 callersFunctionbuild_scaling_rotation
(s, r)
utils/general_utils.py:101
↓ 1 callersFunctioncamera_to_JSON
(id, camera : Camera)
utils/camera_utils.py:62
↓ 1 callersMethodcapture
(self)
scene/gaussian_model.py:88
↓ 1 callersFunctionclustering
(points, num_clusters)
expr_analyze.py:63
↓ 1 callersMethodconstruct_list_of_attributes
(self)
scene/gaussian_model.py:303
↓ 1 callersMethodcreate_custom_mask
Add some cutom masks based on the original FLAME masks
flame_model/flame.py:641
↓ 1 callersMethodcreate_embedding_fn
(self)
utils/triplane_utils.py:104
↓ 1 callersMethodcreate_from_pcd
(self, pcd : Optional[BasicPointCloud], spatial_lr_scale : float)
scene/gaussian_model.py:234
↓ 1 callersFunctioncreate_window
(window_size, channel)
utils/loss_utils.py:27
↓ 1 callersMethoddefine_gui
(self)
utils/viewer_utils.py:246
↓ 1 callersMethoddensify_and_clone
(self, grads, grad_threshold, scene_extent)
scene/gaussian_model.py:578
↓ 1 callersMethoddensify_and_prune
(self, max_grad, min_opacity, extent, max_screen_size)
scene/gaussian_model.py:602
↓ 1 callersMethoddensify_and_split
(self, grads, grad_threshold, scene_extent, N=2)
scene/gaussian_model.py:539
↓ 1 callersMethodembed
(self, inputs)
utils/triplane_utils.py:128
↓ 1 callersFunctionerror_map
(img1, img2)
utils/image_utils.py:22
↓ 1 callersFunctionface_vertices
:param vertices: [batch size, number of vertices, 3] :param faces: [batch size, number of faces, 3] :return: [batch size, number of faces
flame_model/flame.py:56
↓ 1 callersFunctionfarthest_point_sampling
:param points: torch.Tensor, (N, D), D is the dimension of each point :return: (N, )
preprocess_jaw.py:15
↓ 1 callersMethodflush
(self)
utils/general_utils.py:125
↓ 1 callersFunctiongaussian
(window_size, sigma)
utils/loss_utils.py:23
↓ 1 callersFunctiongetProjectionMatrix
(znear, zfar, fovX, fovY)
utils/graphics_utils.py:51
↓ 1 callersFunctionget_center_and_diag
(cam_centers)
scene/dataset_readers.py:55
↓ 1 callersFunctionget_combined_args
(parser : ArgumentParser)
arguments/__init__.py:120
↓ 1 callersMethodget_covariance
(self, scaling_modifier = 1)
scene/gaussian_model.py:224
↓ 1 callersFunctionget_embedder
(multires, input_dims=3)
utils/triplane_utils.py:131
next →1–100 of 250, ranked by callers