Functions — github.com/G-1nOnly/AniSDF

hub / github.com/G-1nOnly/AniSDF / functions

Functions362 in github.com/G-1nOnly/AniSDF

⨍ Functions 362◇ Types & classes 66 ↳ Endpoints 1

↓ 37 callersFunctionget_activation

(name)

models/utils.py:74

↓ 35 callersMethodmean

Return the mean of the embedding weights along a dim.

↓ 28 callersFunctionupdate_module_step

(m, epoch, global_step)

systems/utils.py:349

↓ 27 callersFunctionget_mlp

(n_input_dims, n_output_dims, config)

models/network_utils.py:271

↓ 19 callersFunctionget_rank

()

utils/misc.py:42

↓ 19 callersFunctionscale_anything

(dat, inp_scale, tgt_scale)

models/utils.py:111

↓ 14 callersMethodget_save_path

(self, filename)

utils/mixins.py:33

↓ 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.

datasets/colmap_utils.py:69

↓ 12 callersFunctionget_encoding

(n_input_dims, config)

models/network_utils.py:82

↓ 10 callersFunctioncontract_to_unisphere

(x, radius, contraction_type)

models/geometry.py:19

↓ 9 callersMethodconvert_data

(self, data)

utils/mixins.py:21

↓ 8 callersFunctionconfig_to_primitive

(config, resolve=True)

utils/misc.py:34

↓ 6 callersMethod__init__

(self, config)

models/texture.py:63

↓ 6 callersFunctionchunk_batch

(func, chunk_size, move_to_cpu, *args, **kwargs)

models/utils.py:16

↓ 6 callersMethodsave_image_grid

(self, filename, imgs)

utils/mixins.py:143

↓ 5 callersMethod__init__

(self, dim_in, dim_out, config)

models/network_utils.py:103

↓ 5 callersFunctionget_ray_directions

(W, H, fx, fy, cx, cy, use_pixel_centers=True)

models/ray_utils.py:9

↓ 4 callersFunctionbinary_cross_entropy

F.binary_cross_entropy is not numerically stable in mixed-precision training.

systems/criterions.py:155

↓ 4 callersMethodgeneral_loader

(self, dataset, batch_size)

datasets/dtu.py:182

↓ 4 callersMethodgeneral_loader

(self, dataset, batch_size)

datasets/blender.py:118

↓ 4 callersMethodgeneral_loader

(self, dataset, batch_size)

datasets/shelly.py:123

↓ 4 callersMethodgeneral_loader

(self, dataset, batch_size)

datasets/colmap.py:341

↓ 4 callersFunctionget_rays

(directions, c2w, keepdim=False)

models/ray_utils.py:23

↓ 4 callersMethodpreprocess_data

(self, batch, stage)

systems/base.py:47

↓ 3 callersMethod__init__

(self, resolution, use_torch=True)

models/geometry.py:35

↓ 3 callersMethodget_rgb_image_

(self, img, data_format, data_range)

utils/mixins.py:42

↓ 3 callersMethodmake_linear

(self, dim_in, dim_out, is_first, is_last)

models/network_utils.py:121

↓ 3 callersMethodmake_linear

(self, dim_in, dim_out, is_first, is_last)

models/network_utils.py:186

↓ 2 callersMethod__init__

(self, save_root, version=None, use_version=True)

utils/callbacks.py:17

↓ 2 callersMethod_gaussian

(self, kernel_size, sigma)

systems/criterions.py:85

↓ 2 callersMethod_uniform

(self, kernel_size)

systems/criterions.py:73

↓ 2 callersMethodcreate_kernel

(self, s3im_kernel_size, channel)

systems/s3im.py:25

↓ 2 callersMethodforward_

(self, rays, camera_indices=None)

models/neus.py:234

↓ 2 callersMethodforward_bg_

(self, rays, camera_indices=None)

models/neus.py:167

↓ 2 callersFunctionget_center

(pts)

datasets/colmap.py:20

↓ 2 callersFunctionget_encoding_with_network

(n_input_dims, n_output_dims, encoding_config, network_config)

models/network_utils.py:295

↓ 2 callersMethodget_grayscale_image_

(self, img, data_range, cmap)

utils/mixins.py:86

↓ 2 callersMethodget_uv_image_

(self, img, data_format, data_range, cmap)

utils/mixins.py:59

↓ 2 callersMethodisosurface_

(self, vmin, vmax)

models/geometry.py:85

↓ 2 callersMethodload_state_dict

Loads the schedulers state. Args: state_dict (dict): scheduler state. Should be an object returned from a call to

systems/utils.py:148

↓ 2 callersMethodmake_activation

(self)

models/network_utils.py:142

↓ 2 callersMethodmake_activation

(self)

models/network_utils.py:230

↓ 2 callersFunctionparse_version

(ver)

utils/misc.py:53

↓ 2 callersFunctionquaternion_product

(p, q)

models/ASG_encoder.py:6

↓ 2 callersFunctionread_cameras_binary

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

datasets/colmap_utils.py:107

↓ 2 callersFunctionread_images_binary

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

datasets/colmap_utils.py:167

↓ 2 callersFunctionread_points3d_binary

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

datasets/colmap_utils.py:229

↓ 2 callersMethodregularizations

(self, out)

models/base.py:27

↓ 2 callersMethodsave_img_sequence

(self, filename, img_dir, matcher, save_format='gif', fps=30)

utils/mixins.py:191

↓ 2 callersMethodsave_mesh

(self, filename, v_pos, t_pos_idx, v_tex=None, t_tex_idx=None, v_rgb=None)

utils/mixins.py:211

↓ 2 callersMethodsetup

(self, stage=None)

datasets/dtu.py:168

↓ 2 callersMethodsetup

(self, stage=None)

datasets/blender.py:105

↓ 2 callersMethodsetup

(self, stage=None)

datasets/shelly.py:110

↓ 2 callersMethodsetup

(self, stage=None)

datasets/colmap.py:327

↓ 2 callersFunctionspherical2cartesian

(theta, phi)

models/ASG_encoder.py:42

↓ 1 callersMethod__init__

(self, optimizer, start_factor=1.0 / 3, end_factor=1.0, total_iters=5, last_epoch=-1, verbose

systems/utils.py:255

↓ 1 callersMethod__init__

(self, config, split)

datasets/dtu.py:141

↓ 1 callersMethod__init__

(self, config, split)

datasets/blender.py:78

↓ 1 callersMethod__init__

(self, config, split)

datasets/shelly.py:83

↓ 1 callersMethod__init__

(self, config, split)

datasets/colmap.py:300

↓ 1 callersMethod__init__

(self, in_dim: int, out_dim: int)

models/utils.py:185

↓ 1 callersMethod_create_lattice_grid

(self)

models/geometry.py:140

↓ 1 callersMethod_gaussian_or_uniform_kernel

(self, kernel_size, sigma)

systems/criterions.py:91

↓ 1 callersMethod_get_next_version

(self)

utils/callbacks.py:33

↓ 1 callersMethod_ssim

(self, img1, img2, s3im_kernel, s3im_kernel_size, channel, size_average = True, s3im_stride=None)

systems/s3im.py:31

↓ 1 callersMethodasg_mlp_forward

(self, asg_feature)

models/texture.py:216

↓ 1 callersMethodasg_mlp_forward

(self, asg_feature)

models/texture.py:291

↓ 1 callersMethodasg_mlp_forward

(self, asg_feature)

models/texture.py:390

↓ 1 callersFunctionassoc_legendre_coeff

Compute associated Legendre polynomial coefficients. Returns the coefficient of the cos^k(theta)*sin^m(theta) term in the (l, m)th associated

models/ide_encoder.py:10

↓ 1 callersMethodbuild_nn_modules

(self)

models/utils.py:191

↓ 1 callersFunctionbuild_q

(vec, angle)

models/ASG_encoder.py:30

↓ 1 callersFunctioncartesian2quaternion

(x)

models/ASG_encoder.py:37

↓ 1 callersFunctioncleanup

()

models/utils.py:119

↓ 1 callersFunctioncreate_spheric_poses

(cameras, n_steps=120)

datasets/dtu.py:36

↓ 1 callersFunctioncreate_spheric_poses

(cameras, n_steps=120)

datasets/colmap.py:140

↓ 1 callersFunctiondot

(x, y)

models/utils.py:103

↓ 1 callersFunctiondump_config

(path, config)

utils/misc.py:38

↓ 1 callersMethodexport

(self)

systems/base.py:115

↓ 1 callersMethodexport

(self)

systems/nerf.py:213

↓ 1 callersMethodexport

(self)

systems/neus.py:380

↓ 1 callersMethodexport

(self, export_config)

models/base.py:31

↓ 1 callersMethodforward_

(self, level, threshold=0.)

models/geometry.py:159

↓ 1 callersMethodforward_

(self, rays)

models/nerf.py:61

↓ 1 callersMethodforward_level

(self, points)

models/geometry.py:82

↓ 1 callersMethodforward_level

(self, points)

models/geometry.py:206

↓ 1 callersMethodforward_wo_j

Compute integrated directional encoding (IDE). Args: xyz: [..., 3] array of Cartesian coordinates of directions to evaluate at.

models/ide_encoder.py:132

↓ 1 callersMethodfunc

(self)

systems/criterions.py:8

↓ 1 callersMethodgaussian

(self, s3im_kernel_size, sigma)

systems/s3im.py:21

↓ 1 callersFunctiongen_poses

(basedir, match_type)

scripts/imgs2poses.py:58

↓ 1 callersFunctiongeneralized_binomial_coeff

Compute generalized binomial coefficients.

models/ide_encoder.py:5

↓ 1 callersMethodget_alpha

(self, sdf, normal, dirs, dists)

models/neus.py:119

↓ 1 callersMethodget_file_list

(self)

utils/callbacks.py:56

↓ 1 callersMethodget_image_grid_

(self, imgs)

utils/mixins.py:123

↓ 1 callersFunctionget_ml_array

Create a list with all pairs of (l, m) values to use in the encoding.

models/ide_encoder.py:45

↓ 1 callersFunctionget_parameters

(model, name)

systems/utils.py:305

↓ 1 callersFunctionget_rot

(a, b)

datasets/colmap.py:30

↓ 1 callersFunctionget_scheduler

(name)

systems/utils.py:290

↓ 1 callersFunctiongetattr_recursive

(m, attr)

systems/utils.py:299

↓ 1 callersMethodgrid_vertices

(self)

models/geometry.py:48

next →1–100 of 362, ranked by callers