Functions — github.com/ant-research/TensorialGaussianAvatar

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

Functions250 in github.com/ant-research/TensorialGaussianAvatar

Copied from Plenoxels Continuous learning rate decay function. Adapted from JaxNeRF The returned rate is lr_init when step=0 and lr_fina

utils/general_utils.py:29

↓ 1 callersMethodget_features

(self, cam_pos = None)

scene/gaussian_model.py:189

↓ 1 callersMethodget_fid_by_region

Get face indicies by regions

flame_model/flame.py:896

↓ 1 callersFunctionget_network

(net_type: str)

lpipsPyTorch/modules/networks.py:12

↓ 1 callersFunctionget_state_dict

(net_type: str = 'alex', version: str = '0.1')

lpipsPyTorch/modules/utils.py:11

↓ 1 callersMethodget_vid_except_region

(self, regions)

flame_model/flame.py:882

↓ 1 callersFunctiongrid_sample_wrapper

(grid: torch.Tensor, coords: torch.Tensor, feature_dim , align_corners: bool = True)

utils/triplane_utils.py:11

↓ 1 callersFunctioninit_planes

(grid_dim, in_dim, out_dim, resolution, a, b)

utils/triplane_utils.py:41

↓ 1 callersMethodinitialize_weights

(self)

utils/feature_line.py:58

↓ 1 callersFunctioninterpolate_ms_features

(points, triplane, plane_dim, concat_f, num_levels)

utils/triplane_utils.py:53

↓ 1 callersMethodkeys

(self)

flame_model/flame.py:574

↓ 1 callersFunctionlbs

Performs Linear Blend Skinning with the given shape and pose parameters Parameters ---------- betas : torch.tensor BxNB The tenso

flame_model/lbs.py:101

↓ 1 callersFunctionlength

(x: torch.Tensor, eps: float =1e-20)

utils/vector_ops.py:19

↓ 1 callersFunctionloadCam

(args, id, cam_info, resolution_scale)

utils/camera_utils.py:20

↓ 1 callersMethodload_meshes

(self, train_meshes, test_meshes, tgt_train_meshes, tgt_test_meshes)

scene/flame_gaussian_model.py:115

↓ 1 callersFunctionload_obj

Load a mesh from a .obj file and optionally textures from a .mtl file. Currently this handles verts, faces, vertex texture uv coordinates, no

utils/pytorch3d_load_obj.py:148

↓ 1 callersMethodload_ply

(self, path, **kwargs)

scene/gaussian_model.py:354

↓ 1 callersMethodloss_offset_normalize

(self, timestep)

scene/flame_gaussian_model.py:282

↓ 1 callersFunctionlpips

r"""Function that measures Learned Perceptual Image Patch Similarity (LPIPS). Arguments: x, y (torch.Tensor): the input tensors to co

lpipsPyTorch/__init__.py:6

↓ 1 callersFunctionmkdir_p

(folder_path)

utils/system_utils.py:16

↓ 1 callersFunctionnormalize_activation

(x, eps=1e-10)

lpipsPyTorch/modules/utils.py:6

↓ 1 callersFunctionnullcontext

This is just like contextlib.nullcontext but also works in Python 3.6.

utils/pytorch3d_load_obj.py:53

↓ 1 callersMethodoneupSHdegree

(self)

scene/gaussian_model.py:230

↓ 1 callersMethodorbit_x

(self, angle_x)

utils/viewer_utils.py:172

↓ 1 callersMethodorbit_y

(self, angle_y)

utils/viewer_utils.py:177

↓ 1 callersMethodorbit_z

(self, angle_z)

utils/viewer_utils.py:182

↓ 1 callersMethodprepare_meshes

(self)

scene/flame_gaussian_model.py:187

↓ 1 callersFunctionprepare_output_and_logger

(args)

train.py:203

↓ 1 callersMethodprocess_vertex_mask

Load the vertex masks from the FLAME model and add custom masks

flame_model/flame.py:625

↓ 1 callersFunctionprojection_from_intrinsics

Transform points from camera space (x: right, y: up, z: out) to clip space (x: right, y: up, z: in) Args: K: Intrinsic matrix, (N, 3,

utils/viewer_utils.py:20

↓ 1 callersFunctionpsnr

(img1, img2)

utils/image_utils.py:18

↓ 1 callersFunctionread

()

gaussian_renderer/network_gui.py:43

↓ 1 callersFunctionreadColmapCameras

(cam_extrinsics, cam_intrinsics, images_folder)

scene/dataset_readers.py:77

↓ 1 callersFunctionread_extrinsics_binary

see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBi

scene/colmap_loader.py:180

↓ 1 callersFunctionread_extrinsics_text

Taken from https://github.com/colmap/colmap/blob/dev/scripts/python/read_write_model.py

scene/colmap_loader.py:244

↓ 1 callersFunctionread_intrinsics_binary

see: src/base/reconstruction.cc void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCameras

scene/colmap_loader.py:215

↓ 1 callersFunctionread_intrinsics_text

Taken from https://github.com/colmap/colmap/blob/dev/scripts/python/read_write_model.py

scene/colmap_loader.py:156

↓ 1 callersFunctionread_points3D_binary

see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints

scene/colmap_loader.py:125

↓ 1 callersFunctionread_points3D_text

see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3D

scene/colmap_loader.py:83

↓ 1 callersMethodregister_callbacks

(self)

utils/viewer_utils.py:264

↓ 1 callersFunctionrender_sets

(dataset : ModelParams, iteration : int, pipeline : PipelineParams, skip_train : bool, skip_val : bool, skip_t

render.py:112

↓ 1 callersMethodreplace_tensor_to_optimizer

(self, tensor, name)

scene/gaussian_model.py:408

↓ 1 callersMethodreset

The internal state of the camera is based on the OpenGL convention, but properties are converted to the target convention when queried.

utils/viewer_utils.py:87

↓ 1 callersMethodreset_opacity

(self)

scene/gaussian_model.py:349

↓ 1 callersMethodrestore

(self, model_args, training_args)

scene/gaussian_model.py:114

↓ 1 callersMethodsave_ply

(self, path)

scene/gaussian_model.py:321

↓ 1 callersMethodscale

(self, delta)

utils/viewer_utils.py:187

↓ 1 callersFunctionsearchForMaxIteration

(folder)

utils/system_utils.py:26

↓ 1 callersMethodsetup_functions

(self)

scene/gaussian_model.py:29

↓ 1 callersFunctionstrip_lowerdiag

(L)

utils/general_utils.py:64

↓ 1 callersFunctionstrip_symmetric

(sym)

utils/general_utils.py:75

↓ 1 callersFunctiontraining

(dataset, opt, pipe, testing_iterations, saving_iterations, checkpoint_iterations, checkpoint, debug_from)

train.py:35

↓ 1 callersFunctiontraining_report

(tb_writer, iteration, losses, elapsed, testing_iterations, scene : Scene, renderFunc, renderArgs)

train.py:225

↓ 1 callersFunctiontransform_mat

Creates a batch of transformation matrices Args: - R: Bx3x3 array of a batch of rotation matrices - t: Bx3x1 array of a batch of t

flame_model/lbs.py:242

↓ 1 callersMethodupdate_cano_mesh

(self)

scene/flame_gaussian_model.py:88

↓ 1 callersMethodupdate_learning_rate

Learning rate scheduling per step

scene/gaussian_model.py:295

↓ 1 callersFunctionvertices2joints

Calculates the 3D joint locations from the vertices Parameters ---------- J_regressor : torch.tensor JxV The regressor array that

flame_model/lbs.py:198

↓ 1 callersFunctionvertices2landmarks

Calculates landmarks by barycentric interpolation Parameters ---------- vertices: torch.tensor BxVx3, dtype = torch.float32 The t

flame_model/lbs.py:60

↓ 1 callersMethodz_score

(self, x: torch.Tensor)

lpipsPyTorch/modules/networks.py:50

Method__del__

(self)

utils/viewer_utils.py:243

(self, sh_degree : int, static: str, dynamic: str)

scene/gaussian_model.py:47

Method__init__

(self, cameras: List[Camera])

scene/__init__.py:32

Method__init__

b :param path: Path to colmap scene main folder.

scene/__init__.py:73

Method__init__

(self, colmap_id, R, T, FoVx, FoVy, bg, image_width, image, image_height, image_path, image_n

scene/cameras.py:18

Method__init__

(self, width, height, fovy, fovx, znear, zfar, world_view_transform, full_proj_transform, timestep)

scene/cameras.py:50

Method__init__

(self, source_path, sh_degree : int, disable_flame_static_offset=False, not_finetune_flame_params=False, \

scene/flame_gaussian_model.py:49

Method__init__

(self, **kwargs)

utils/triplane_utils.py:100

Method__init__

(self, spatial_bounds, grid_dim=2, in_dim=3, out_dim=64, resolution=[128, 128, 128], a=-0.1, b=0.1)

utils/triplane_utils.py:146

Method__init__

(self, expr_num = 80, key_jaw_pose_num = 16, line_size = (64, 64, 64), feat_dim = [32, 32, 32], n_layer = 2, n

utils/feature_line.py:27

Method__init__

(self, silent)

utils/general_utils.py:115

Method__init__

(self, W, H, r=2, fovy=60, znear=0.01, zfar=10, convention: Literal["opengl", "opencv"]="opengl", save_path='c

utils/viewer_utils.py:74

Method__init__

(self, cfg: Mini3DViewerConfig, title='Mini3DViewer')

utils/viewer_utils.py:207

Method__init__

( self, shape_params, expr_params, flame_model_path=FLAME_MODEL_PATH,

flame_model/flame.py:83

Method__init__

(self)

flame_model/flame.py:561

Method__init__

( self, flame_parts_path=FLAME_PARTS_PATH, faces=None, face

flame_model/flame.py:581

Method__init__

(self, parser, sentinel=False)

arguments/__init__.py:48

Method__init__

(self, parser)

arguments/__init__.py:77

Method__init__

(self, parser)

arguments/__init__.py:84

Method__init__

(self, n_channels_list: Sequence[int])

lpipsPyTorch/modules/networks.py:24

Method__init__

(self)

lpipsPyTorch/modules/networks.py:37

Method__init__

(self)

lpipsPyTorch/modules/networks.py:67

Method__init__

(self)

lpipsPyTorch/modules/networks.py:78

Method__init__

(self, net_type: str = 'alex', version: str = '0.1')

lpipsPyTorch/modules/lpips.py:17

Method__iter__

(self)

flame_model/flame.py:570

flame_model/flame.py:564

Methodbuild_covariance_from_scaling_rotation

(scaling, scaling_modifier, rotation)

scene/gaussian_model.py:30

Methodcallback_key_press

(sender, app_data)

utils/viewer_utils.py:340

Methodcallback_mouse_button_down

(sender, app_data)

utils/viewer_utils.py:319

Methodcallback_mouse_move

(sender, app_data)

utils/viewer_utils.py:281

Methodcallback_mouse_release

(sender, app_data)

utils/viewer_utils.py:327

Methodcallback_mouse_wheel

(sender, app_data)

utils/viewer_utils.py:334

Methodcallback_resize

(sender, app_data)

utils/viewer_utils.py:265

Methodclear

(self)

utils/viewer_utils.py:114

Functioncompute_face_normals

(verts, faces)

utils/graphics_utils.py:105

Functioncompute_vertex_normals

(verts, faces)

utils/graphics_utils.py:137

Functionembed

(x, eo=embedder_obj)

utils/triplane_utils.py:142

Functioneval_sh

Evaluate spherical harmonics at unit directions using hardcoded SH polynomials. Works with torch/np/jnp. ... Can be 0 or more batch d

utils/sh_utils.py:57

← previous next →101–200 of 250, ranked by callers