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

↓ 2 callersFunctiontransform_pose_llff

(poses)

src/data/data_util/llff_pair.py:300

↓ 2 callersMethodupdate_config

(self, stage)

src/model/dvgo/model.py:724

↓ 2 callersFunctionviewmatrix

(z, up, pos)

src/data/data_util/llff.py:158

↓ 2 callersFunctionviewmatrix

(z, up, pos)

src/data/data_util/nerf_360_v2.py:151

↓ 2 callersFunctionviewmatrix

(z, up, pos)

src/data/data_util/refnerf_real.py:149

↓ 2 callersFunctionviewmatrix

(z, up, pos)

src/data/data_util/llff_single.py:61

↓ 2 callersFunctionviewmatrix

(z, up, pos)

src/data/data_util/llff_pair.py:172

↓ 2 callersMethodvolume_render_fused

Standard volume rendering with fused MSE gradient generation, given a ground truth color for each pixel. Will update the

src/model/plenoxel/sparse_grid.py:362

↓ 1 callersFunctionHE

(img)

utils/Histogram_Equalization.py:20

↓ 1 callersFunctionR_axis_angle

Generate the rotation matrix from the axis-angle notation. Conversion equations ==================== From Wikipedia (http://en.wikipedia.o

src/data/pose_utils.py:155

↓ 1 callersFunctionR_to_axis_angle

Convert the rotation matrix into the axis-angle notation. Conversion equations ==================== From Wikipedia (http://en.wikipedia.or

src/data/pose_utils.py:210

↓ 1 callersMethod__init__

(self, rays_info)

src/data/interface.py:315

↓ 1 callersMethod_get_data_grads

(self)

src/model/plenoxel/sparse_grid.py:1297

↓ 1 callersMethod_get_rand_cells_background

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

src/model/plenoxel/sparse_grid.py:1391

↓ 1 callersMethod_get_sparse_background_grad_indexer

(self)

src/model/plenoxel/sparse_grid.py:1331

↓ 1 callersMethod_get_sparse_grad_indexer

(self)

src/model/plenoxel/sparse_grid.py:1315

↓ 1 callersFunction_load_data

(basedir, factor=None, width=None, height=None, load_imgs=True)

src/data/data_util/llff.py:78

↓ 1 callersFunction_load_data

(basedir, factor=None, width=None, height=None, load_imgs=True)

src/data/data_util/nerf_360_v2.py:78

↓ 1 callersFunction_load_data

(basedir, factor=None, width=None, height=None, load_imgs=True)

src/data/data_util/refnerf_real.py:76

↓ 1 callersFunction_load_data

(basedir, scene_name, factor=None, width=None, height=None, load_imgs=True)

src/data/data_util/llff_single.py:20

↓ 1 callersFunction_load_data

(basedir, factor=None, width=None, height=None, load_imgs=True)

src/data/data_util/llff_pair.py:78

↓ 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

src/model/refnerf/ref_utils.py:65

↓ 1 callersMethodbackward

(ctx, grad_out)

src/model/plenoxel/autograd.py:90

↓ 1 callersMethodbackward

(ctx, grad_back)

src/model/dvgo/dcvgo.py:511

↓ 1 callersFunctionbatchified_get_rays

( intrinsics, extrinsics, image_sizes, use_pixel_centers, get_radii, ndc_coord, nd

src/data/ray_utils.py:30

↓ 1 callersMethodcompute_bbox_by_coarse_geo

(self, model_class, model_path, thres)

src/model/dvgo/model.py:574

↓ 1 callersMethodcompute_normal_mae

(self, rendered_results, normals)

src/model/refnerf/model.py:497

↓ 1 callersFunctioncompute_tensorf_feat

( xy_plane, xz_plane, yz_plane, x_vec, y_vec, z_vec, f_vec, ind_norm )

src/model/dvgo/grid.py:263

↓ 1 callersFunctioncompute_tensorf_val

(xy_plane, xz_plane, yz_plane, x_vec, y_vec, z_vec, ind_norm)

src/model/dvgo/grid.py:322

↓ 1 callersFunctioncompute_tesselation_weights

Tesselate the vertices of a triangle by a factor of `v`.

src/model/refnerf/helper.py:277

↓ 1 callersFunctionconvertPNG

(pngfile,outdir)

depth_trans.py:14

↓ 1 callersFunctionconvert_to_ndc

Convert a set of rays to NDC coordinates.

src/data/ray_utils.py:10

↓ 1 callersMethoddensity_total_variation_add_grad

(self, weight, dense_mode)

src/model/dvgo/dvgo.py:358

↓ 1 callersFunctionexpected_sin

(x, x_var)

src/model/mipnerf/helper.py:185

↓ 1 callersFunctionexpected_sin

(x, x_var)

src/model/refnerf/helper.py:189

↓ 1 callersFunctionfocus_point_fn

Calculate nearest point to all focal axes in poses.

src/data/pose_utils.py:52

↓ 1 callersFunctiongeneralized_binomial_coeff

Compute generalized binomial coefficients.

src/model/refnerf/ref_utils.py:59

↓ 1 callersMethodgenerate_camera_list

( self, intrinsics=None, extrinsics=None, ndc_coeffs=None, image_size=None )

src/model/plenoxel/model.py:266

↓ 1 callersFunctiongenerate_ide_fn

Generate integrated directional encoding (IDE) function. This function returns a function that computes the integrated directional encoding f

src/model/refnerf/ref_utils.py:114

↓ 1 callersFunctionget_ml_array

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

src/model/refnerf/ref_utils.py:101

↓ 1 callersFunctionget_positions

(theta)

src/data/pose_utils.py:97

↓ 1 callersMethodglobal_average_pool

(self, x)

src/model/aleth_nerf/helper.py:106

↓ 1 callersMethodglobal_average_pool

(self, x)

src/model/aleth_nerf_exp/helper.py:106

↓ 1 callersMethodhit_coarse_geo

Check whether the rays hit the solved coarse geometry or not

src/model/dvgo/dvgo.py:373

↓ 1 callersFunctionimread

(f)

src/data/data_util/llff.py:139

↓ 1 callersFunctionimread

(f)

src/data/data_util/nerf_360_v2.py:135

↓ 1 callersFunctionimread

(f)

src/data/data_util/refnerf_real.py:133

↓ 1 callersFunctioninit

()

src/model/dvgo/__global__.py:25

↓ 1 callersFunctioninit_lpips

(net_name, device)

src/model/dvgo/utils.py:148

↓ 1 callersMethodinplace_l2_color_grad

Add gradient of L2 regularization for color directly into the gradient tensor, multiplied by 'scaling' (no CUDA extension use

src/model/plenoxel/sparse_grid.py:1015

↓ 1 callersMethodinplace_tv_background_grad

Add gradient of total variation for color directly into the gradient tensor, multiplied by 'scaling'

src/model/plenoxel/sparse_grid.py:1053

↓ 1 callersMethodinplace_tv_color_grad

Add gradient of total variation for color directly into the gradient tensor, multiplied by 'scaling' :param start_dim: int,

src/model/plenoxel/sparse_grid.py:890

↓ 1 callersMethodinplace_tv_grad

Add gradient of total variation for sigma as in Neural Volumes [Lombardi et al., ToG 2019] directly into the gradient tensor,

src/model/plenoxel/sparse_grid.py:835

↓ 1 callersMethodinplace_tv_lumisphere_grad

( self, grad: torch.Tensor, start_dim: int = 0, end_dim: Optional[int] = None,

src/model/plenoxel/sparse_grid.py:957

↓ 1 callersFunctionintegrated_dir_enc_fn

Function returning integrated directional encoding (IDE). Args: xyz: [..., 3] array of Cartesian coordinates of directions to eva

src/model/refnerf/ref_utils.py:145

↓ 1 callersMethodinward_nearfar_heuristic

(self, cam_o, ratio=0.05)

src/data/interface.py:95

↓ 1 callersFunctionis_pow2

(x: int)

src/model/plenoxel/utils.py:47

↓ 1 callersMethodk0_total_variation_add_grad

(self, weight, dense_mode)

src/model/dvgo/dvgo.py:362

↓ 1 callersFunctionload_blender_data

( datadir: str, scene_name: str, compare_method: str, train_skip: int, val_skip: int,

src/data/data_util/blender.py:53

↓ 1 callersFunctionload_blender_ms_data

( datadir: str, scene_name: str, train_skip: int, val_skip: int, test_skip: int, cam_s

src/data/data_util/blender_ms.py:52

↓ 1 callersFunctionload_lf_data

( datadir: str, scene_name: str, train_skip: int, val_skip: int, test_skip: int, cam_s

src/data/data_util/lf.py:87

↓ 1 callersFunctionload_llff_data

( datadir: str, scene_name: str, factor: int, ndc_coord: bool, # True recenter: bool,

src/data/data_util/llff.py:295

↓ 1 callersFunctionload_llff_data_pair

( datadir: str, scene_name: str, factor: int, ndc_coord: bool, # True recenter: bool,

src/data/data_util/llff_pair.py:309

↓ 1 callersFunctionload_nerf_360_v2_data

( datadir: str, scene_name: str, factor: int, cam_scale_factor: float, train_skip: int,

src/data/data_util/nerf_360_v2.py:346

↓ 1 callersFunctionload_refnerf_real_data

( datadir: str, scene_name: str, factor: int, cam_scale_factor: float, train_skip: int,

src/data/data_util/refnerf_real.py:342

↓ 1 callersFunctionload_shiny_blender_data

( datadir: str, scene_name: str, train_skip: int, val_skip: int, test_skip: int, cam_s

src/data/data_util/shiny_blender.py:53

↓ 1 callersFunctionload_tnt_data

( datadir: str, scene_name: str, train_skip: int, val_skip: int, test_skip: int, cam_s

src/data/data_util/tnt.py:87

↓ 1 callersMethodmaskout_near_cam_vox

(self, cam_o, near_clip)

src/model/dvgo/dvgo.py:209

↓ 1 callersFunctionmin_line_dist

(rays_o, rays_d)

src/data/data_util/llff.py:224

↓ 1 callersFunctionmin_line_dist

(rays_o, rays_d)

src/data/data_util/nerf_360_v2.py:217

↓ 1 callersFunctionmin_line_dist

(rays_o, rays_d)

src/data/data_util/refnerf_real.py:215

↓ 1 callersFunctionmin_line_dist

(rays_o, rays_d)

src/data/data_util/llff_single.py:127

↓ 1 callersFunctionmin_line_dist

(rays_o, rays_d)

src/data/data_util/llff_pair.py:238

↓ 1 callersFunctionmorton_code_3

(x, y, z)

src/model/plenoxel/utils.py:69

↓ 1 callersMethodon_train_batch_end

(self, trainer, pl_module, outputs, batch, batch_idx)

src/model/dvgo/model.py:33

↓ 1 callersMethodoptim_background_step

Execute RMSprop or sgd step on density

src/model/plenoxel/sparse_grid.py:1158

↓ 1 callersMethodoptim_density_step

Execute RMSprop or sgd step on density

src/model/plenoxel/sparse_grid.py:1083

↓ 1 callersMethodoptim_sh_step

Execute RMSprop/SGD step on SH

src/model/plenoxel/sparse_grid.py:1123

↓ 1 callersMethodorientation_loss

(self, rendered_results, viewdirs)

src/model/refnerf/model.py:461

↓ 1 callersFunctionpad_poses

Pad [..., 3, 4] pose matrices with a homogeneous bottom row [0,0,0,1].

src/data/pose_utils.py:4

↓ 1 callersFunctionpose_spherical

(theta, phi, radius)

src/data/data_util/shiny_blender.py:42

↓ 1 callersFunctionpose_spherical

(theta, phi, radius)

src/data/data_util/blender_ms.py:41

↓ 1 callersFunctionposes_avg

(poses)

src/data/data_util/nerf_360_v2.py:165

↓ 1 callersFunctionposes_avg

(poses)

src/data/data_util/refnerf_real.py:163

↓ 1 callersMethodpredict

(samples, mlp)

src/model/nerfpp/model.py:199

↓ 1 callersMethodpredicted_normal_loss

(self, rendered_results)

src/model/refnerf/model.py:479

↓ 1 callersFunctionrecenter_poses

(poses)

src/data/data_util/llff.py:200

↓ 1 callersFunctionrecenter_poses

(poses)

src/data/data_util/llff_single.py:103

↓ 1 callersFunctionrecenter_poses

(poses)

src/data/data_util/llff_pair.py:214

↓ 1 callersFunctionrender_path_spiral

(c2w, up, rads, focal, zdelta, zrate, rots, N)

src/data/data_util/llff.py:184

↓ 1 callersFunctionrender_path_spiral

(c2w, up, rads, focal, zdelta, zrate, rots, N)

src/data/data_util/llff_single.py:87

↓ 1 callersFunctionrender_path_spiral

(c2w, up, rads, focal, zdelta, zrate, rots, N)

src/data/data_util/llff_pair.py:198

↓ 1 callersMethodresample

Resample and sparsify the grid; used to increase the resolution :param reso: int or List[int, int, int], resolution for resampled gri

src/model/plenoxel/sparse_grid.py:456

↓ 1 callersFunctionrun

( ginc: str, ginb: str, resume_training: bool, ckpt_path: Optional[str], scene_name: Optio

run.py:39

↓ 1 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/dvgo.py:397

↓ 1 callersFunctionselect_callback

(model_name)

utils/select_option.py:90

↓ 1 callersFunctionselect_dataset

( dataset_name: str, datadir: str, scene_name: str, )

utils/select_option.py:56

↓ 1 callersFunctionselect_model

( model_name: str, eta:float, con:float )

utils/select_option.py:31

↓ 1 callersMethodset_pervoxel_lr

(self, count)

src/model/dvgo/masked_adam.py:39

↓ 1 callersMethodsetup

(self, stage)

src/model/dvgo/model.py:335

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