Functions — github.com/cuiziteng/Aleth-NeRF

hub / github.com/cuiziteng/Aleth-NeRF / functions

Functions605 in github.com/cuiziteng/Aleth-NeRF

⨍ Functions 605◇ Types & classes 78

↓ 35 callersMethodalter_gather_cat

(self, outputs, key, image_sizes)

src/model/interface.py:26

↓ 12 callersMethod_to_cpp

Generate object to pass to C++

src/model/plenoxel/sparse_grid.py:1262

↓ 12 callersFunctionnormalize

(x)

src/data/data_util/llff.py:154

↓ 12 callersFunctionnormalize

(x)

src/data/data_util/llff_single.py:57

↓ 12 callersFunctionnormalize

(x)

src/data/data_util/llff_pair.py:168

↓ 11 callersFunctionnormalize

(x)

src/data/data_util/nerf_360_v2.py:147

↓ 11 callersFunctionnormalize

(x)

src/data/data_util/refnerf_real.py:145

↓ 11 callersMethodtrain_dataloader

(self)

src/data/interface.py:193

↓ 10 callersFunctionfind_files

(dir, exts)

src/data/data_util/tnt.py:19

↓ 10 callersFunctionfind_files

(dir, exts)

src/data/data_util/lf.py:19

↓ 9 callersMethod__init__

( self, datadir: str, scene_name: str, # LF specific cam_scale_factor:

src/data/litdata.py:264

↓ 9 callersMethodlpips_each

(self, preds, gts)

src/model/interface.py:83

↓ 9 callersMethodpsnr_each

(self, preds, gts)

src/model/interface.py:60

↓ 9 callersMethodssim_each

(self, preds, gts)

src/model/interface.py:71

↓ 8 callersMethod_fetch_links

(self, links)

src/model/plenoxel/sparse_grid.py:228

↓ 8 callersMethodlpips

(self, preds, gts, i_train, i_val, i_test)

src/model/interface.py:125

↓ 8 callersMethodpsnr

(self, preds, gts, i_train, i_val, i_test)

src/model/interface.py:95

↓ 8 callersMethodssim

(self, preds, gts, i_train, i_val, i_test)

src/model/interface.py:110

↓ 8 callersMethodwrite_stats

(self, fpath, *stats)

src/model/interface.py:139

↓ 7 callersMethodclone

(self)

src/model/plenoxel/utils.py:199

↓ 7 callersMethodstep

(self, closure=None)

src/model/dvgo/masked_adam.py:44

↓ 6 callersMethodget_dense_grid

(self)

src/model/dvgo/grid.py:85

↓ 6 callersMethodtotal_variation_add_grad

Add gradients by total variation loss in-place

src/model/dvgo/grid.py:77

↓ 5 callersMethod__init__

(self, batch_size, num_replicas, rank, tpu)

src/data/sampler.py:15

↓ 5 callersFunctionto8b

(x)

utils/store_image.py:15

↓ 4 callersFunction_index_across_sides

(nidx: CubemapCoord, uori, vori, mu, mv)

src/model/plenoxel/utils.py:307

↓ 4 callersMethod_maybe_convert_sparse_grad_indexer

Automatically convert sparse grad indexer from mask to indices, if it is efficient

src/model/plenoxel/sparse_grid.py:1339

↓ 4 callersMethodactivate_density

(self, density, interval=None)

src/model/dvgo/dvgo.py:366

↓ 4 callersMethodget_expon_lr_func

( self, lr_init, lr_final, lr_delay_steps, lr_delay_mult, max_steps )

src/model/plenoxel/model.py:286

↓ 4 callersMethodget_render_kwargs

( self, )

src/model/dvgo/model.py:807

↓ 4 callersMethodquery_in

(self, cubemap: torch.Tensor)

src/model/plenoxel/utils.py:188

↓ 4 callersMethodsplit_each

( self, _images, _normals, render_poses, idx, dummy=True,

src/data/interface.py:103

↓ 3 callersFunction_expand_bits

(v)

src/model/plenoxel/utils.py:52

↓ 3 callersFunction_get_c_extension

()

src/model/plenoxel/__global__.py:16

↓ 3 callersMethod_get_rand_cells

( self, sparse_frac: float, force: bool = False, contiguous: bool = True )

src/model/plenoxel/sparse_grid.py:1359

↓ 3 callersMethod_grid_size

(self)

src/model/plenoxel/sparse_grid.py:1294

↓ 3 callersFunction_minify

(basedir, factors=[], resolutions=[])

src/data/data_util/llff.py:20

↓ 3 callersFunction_minify

(basedir, factors=[], resolutions=[])

src/data/data_util/nerf_360_v2.py:20

↓ 3 callersFunction_minify

(basedir, factors=[], resolutions=[])

src/data/data_util/refnerf_real.py:18

↓ 3 callersFunction_minify

(basedir, factors=[], resolutions=[])

src/data/data_util/llff_pair.py:20

↓ 3 callersFunction_unexpand_bits

(v)

src/model/plenoxel/utils.py:60

↓ 3 callersMethodactivate_density

(self, density, interval=None)

src/model/dvgo/dcvgo.py:293

↓ 3 callersMethodactivate_density

(self, density, interval=None)

src/model/dvgo/dmpigo.py:308

↓ 3 callersMethodbackward

alpha' = interval * ((1 + exp(density + shift)) ^ (-interval-1)) * exp(density + shift)' = interval * ((1 + exp(density + shif

src/model/dvgo/dvgo.py:563

↓ 3 callersFunctioncompute_sq_dist

Compute the squared Euclidean distance between all pairs of columns.

src/model/refnerf/helper.py:265

↓ 3 callersFunctionnormalize

Normalization helper function.

src/data/pose_utils.py:70

↓ 3 callersFunctionpose_spherical

(theta, phi, radius)

src/data/data_util/blender.py:42

↓ 3 callersFunctionposes_avg

(poses)

src/data/data_util/llff.py:172

↓ 3 callersFunctionposes_avg

(poses)

src/data/data_util/llff_single.py:75

↓ 3 callersFunctionposes_avg

(poses)

src/data/data_util/llff_pair.py:186

↓ 3 callersMethodsample

Grid sampling with trilinear interpolation. Behaves like torch.nn.functional.grid_sample with padding mode border and align_c

src/model/plenoxel/sparse_grid.py:243

↓ 3 callersMethodscene_rep_reconstruction

(self, xyz_min, xyz_max, stage, coarse_ckpt_path=None)

src/model/dvgo/model.py:599

↓ 2 callersMethod__init__

( self, num_levels: int = 2, # coarse_mlp and fine_mlp min_deg_point: int = 0,

src/model/nerf/model.py:114

↓ 2 callersMethod__init__

( self, num_samples: int = 128, num_levels: int = 2, resample_padding: float =

src/model/mipnerf/model.py:115

↓ 2 callersMethod__init__

( self, num_samples: int = 128, num_levels: int = 2, resample_padding: float =

src/model/refnerf/model.py:225

↓ 2 callersMethod__init__

( self, num_levels: int = 2, min_deg_point: int = 0, max_deg_point: int = 10,

src/model/nerfpp/model.py:113

↓ 2 callersMethod__init__

( self, min_deg_point, max_deg_point, deg_view, netdepth: int = 8,

src/model/aleth_nerf/model.py:32

↓ 2 callersMethod__init__

(self, patch_size=64, mean_val=0.2)

src/model/aleth_nerf/helper.py:82

↓ 2 callersMethod__init__

( self, min_deg_point, max_deg_point, deg_view, netdepth: int = 8,

src/model/aleth_nerf_exp/model.py:32

↓ 2 callersMethod__init__

(self, patch_size=64, mean_val=0.2)

src/model/aleth_nerf_exp/helper.py:82

↓ 2 callersMethod__init__

( self, path=None, mask_cache_thres=None, mask=None, xyz_min=None, xyz_max=None )

src/model/dvgo/grid.py:381

↓ 2 callersMethod_det

(self, r: float)

src/model/plenoxel/utils.py:439

↓ 2 callersMethod_get_sparse_sh_grad_indexer

(self)

src/model/plenoxel/sparse_grid.py:1323

↓ 2 callersFunction_index_across_one_side

(mask, d, ori, other_coord)

src/model/plenoxel/utils.py:315

↓ 2 callersMethod_set_grid_resolution

(self, num_voxels)

src/model/dvgo/dcvgo.py:159

↓ 2 callersMethod_set_grid_resolution

(self, num_voxels)

src/model/dvgo/dvgo.py:178

↓ 2 callersMethod_set_grid_resolution

(self, num_voxels, mpi_depth)

src/model/dvgo/dmpigo.py:171

↓ 2 callersMethodaccelerate

Accelerate

src/model/plenoxel/sparse_grid.py:730

↓ 2 callersFunctioncast_rays

(t_vals, origins, directions)

src/model/nerf/helper.py:25

↓ 2 callersFunctioncast_rays

(t_vals, origins, directions, radii, ray_shape)

src/model/mipnerf/helper.py:239

↓ 2 callersFunctioncast_rays

(t_vals, origins, directions, radii, ray_shape)

src/model/refnerf/helper.py:243

↓ 2 callersFunctioncast_rays

(t_vals, origins, directions)

src/model/nerfpp/helper.py:24

↓ 2 callersFunctioncast_rays

(t_vals, origins, directions)

src/model/aleth_nerf/helper.py:115

↓ 2 callersFunctioncast_rays

(t_vals, origins, directions)

src/model/aleth_nerf_exp/helper.py:115

↓ 2 callersFunctionconvolve2d

(z, f)

src/model/dvgo/utils.py:112

↓ 2 callersFunctioncreate_full_step_id

(shape)

src/model/dvgo/dmpigo.py:451

↓ 2 callersFunctiondepth2pts_outside

Compute the points along the ray that are outside of the unit sphere. Args: rays_o: [num_rays, 3]. Ray origins of the points. rays

src/model/nerfpp/helper.py:192

↓ 2 callersMethodget_kwargs

(self)

src/model/dvgo/dvgo.py:189

↓ 2 callersMethodget_train_kwargs

( self, )

src/model/dvgo/model.py:777

↓ 2 callersFunctionimread

(f)

src/data/data_util/llff_single.py:44

↓ 2 callersFunctionimread

(f)

src/data/data_util/llff_pair.py:145

↓ 2 callersFunctionlift_gaussian

(d, t_mean, t_var, r_var)

src/model/mipnerf/helper.py:192

↓ 2 callersFunctionlift_gaussian

(d, t_mean, t_var, r_var)

src/model/refnerf/helper.py:196

↓ 2 callersFunctionload_image_data

( datadir: str, scene_name: str, ndc_coord: bool=True, # True recenter: bool=True, bd_f

src/data/data_util/llff_single.py:198

↓ 2 callersFunctionparse_txt

(filename)

src/data/data_util/tnt.py:100

↓ 2 callersFunctionparse_txt

(filename)

src/data/data_util/lf.py:100

↓ 2 callersMethodrender_rays

(self, batch, batch_idx)

src/model/nerf/model.py:239

↓ 2 callersMethodrender_rays

(self, batch, batch_idx)

src/model/mipnerf/model.py:249

↓ 2 callersMethodrender_rays

( self, batch, batch_idx, prefix="", cpu=False, randomize=Fals

src/model/plenoxel/model.py:409

↓ 2 callersMethodrender_rays

(self, batch, batch_idx)

src/model/refnerf/model.py:366

↓ 2 callersMethodrender_rays

(self, batch, batch_idx)

src/model/nerfpp/model.py:286

↓ 2 callersMethodrender_rays

(self, batch, batch_idx)

src/model/aleth_nerf/model.py:331

↓ 2 callersMethodrender_rays

(self, batch, batch_idx)

src/model/aleth_nerf_exp/model.py:327

↓ 2 callersMethodrender_rays

(self, batch, batch_idx)

src/model/dvgo/model.py:510

↓ 2 callersMethodsample_ray

Sample query points on rays. All the output points are sorted from near to far. Input: rays_o, rayd_d: both in [N, 3] in

src/model/dvgo/dcvgo.py:300

↓ 2 callersMethodsample_ray

Sample query points on rays. All the output points are sorted from near to far. Input: rays_o, rayd_d: both in [N, 3] in

src/model/dvgo/dmpigo.py:313

↓ 2 callersMethodscale_volume_grid

(self, new_world_size)

src/model/dvgo/grid.py:64

↓ 2 callersFunctiontesselate_geodesic

Tesselate the vertices of a geodesic polyhedron. Args: base_verts: tensor of floats, the vertex coordinates of the geodesic. base

src/model/refnerf/helper.py:290

↓ 2 callersFunctiontransform_pose_llff

(poses)

src/data/data_util/llff.py:286

↓ 2 callersFunctiontransform_pose_llff

(poses)

src/data/data_util/llff_single.py:189

next →1–100 of 605, ranked by callers