Functions — github.com/cuiziteng/Aleth-NeRF

CodeHub MCPcopy Create free account

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

Functions605 in github.com/cuiziteng/Aleth-NeRF

⨍ Functions 605◇ Types & classes 78

Methodget_dense_grid

src/model/dvgo/grid.py:234

Methodget_kwargs

src/model/dvgo/dcvgo.py:171

Methodget_kwargs

src/model/dvgo/dmpigo.py:185

Methodgrid2world

Grid coordinates to world coordinates. Grid coordinates are normalized to [0, n_voxels] in each side :param points: (N, 3)

src/model/plenoxel/sparse_grid.py:754

src/model/plenoxel/model.py:289

Functionimg2mse

src/model/nerf/helper.py:17

Functionimg2mse

src/model/mipnerf/helper.py:15

Functionimg2mse

src/model/plenoxel/utils.py:20

Functionimg2mse

src/model/refnerf/helper.py:17

Functionimg2mse

src/model/nerfpp/helper.py:16

Functionimg2mse

src/model/aleth_nerf/helper.py:10

Functionimg2mse

src/model/aleth_nerf_exp/helper.py:10

Functionimg2mse_gamma

(x, y, gamma=2, type='under')

src/model/aleth_nerf/helper.py:14

Functionimg2mse_gamma

(x, y, gamma=2, type='under')

src/model/aleth_nerf_exp/helper.py:14

Functionimg2mse_tone

src/model/aleth_nerf/helper.py:23

Functionimg2mse_tone

src/model/aleth_nerf_exp/helper.py:23

Functioninfer_n_samples

lib/dvgo/render_utils.cpp:60

Functioninfer_n_samples

lib/dvgo/cuda/render_utils.cpp:60

Functioninfer_ray_start_dir

lib/dvgo/render_utils.cpp:67

Functioninfer_ray_start_dir

lib/dvgo/cuda/render_utils.cpp:67

Functioninfer_t_minmax

lib/dvgo/render_utils.cpp:50

Functioninfer_t_minmax

lib/dvgo/cuda/render_utils.cpp:50

Functioninit_weights

src/model/plenoxel/utils.py:494

Functionintegrated_pos_enc

(samples, min_deg, max_deg)

src/model/mipnerf/helper.py:134

Functionintegrated_pos_enc

(means, covs, min_deg, max_deg)

src/model/refnerf/helper.py:138

Methodintersect

Find far intersection of all rays with sphere of radius r

src/model/plenoxel/utils.py:415

Methodintersect_near

Find near intersection of all rays with sphere of radius r

src/model/plenoxel/utils.py:427

Functionintersect_sphere

Compute the depth of the intersection point between this ray and unit sphere. Args: rays_o: [num_rays, 3]. Ray origins. rays_d: [n

src/model/nerfpp/helper.py:170

Functioninthroot

(x: int, n: int)

src/model/plenoxel/utils.py:28

Functioninv_morton_code_3

src/model/plenoxel/utils.py:76

Methodis_cubic_pow2

Check if the current grid is cubic (same in all dims) with power-of-2 size. This allows for conversion to svox 1 and Z-order curve (M

src/model/plenoxel/sparse_grid.py:1254

src/model/plenoxel/dataclass.py:89

src/model/plenoxel/dataclass.py:142

Methodk0_total_variation_add_grad

(self, weight, dense_mode)

src/model/dvgo/dcvgo.py:289

Methodk0_total_variation_add_grad

(self, weight, dense_mode)

src/model/dvgo/dmpigo.py:303

Functionl2_normalize

Normalize x to unit length along last axis.

src/model/refnerf/ref_utils.py:35

Functionlift_and_diagonalize

Project `mean` and `cov` onto basis and diagonalize the projected cov.

src/model/refnerf/helper.py:257

Functionlinear_to_srgb

(linear, eps=1e-10)

src/model/refnerf/helper.py:25

Functionload_checkpoint

(model, optimizer, ckpt_path, no_reload_optimizer)

src/model/dvgo/utils.py:69

Functionload_model

(model_class, ckpt_path)

src/model/dvgo/utils.py:78

Functionmaskcache_lookup

lib/dvgo/render_utils.cpp:109

Functionmaskcache_lookup

lib/dvgo/cuda/render_utils.cpp:109

Functionmasked_adam_upd

lib/dvgo/adam_upd.cpp:50

Functionmasked_adam_upd

lib/dvgo/cuda/adam_upd.cpp:50

(device="cuda")

src/model/plenoxel/utils.py:443

Functionmse2psnr

src/model/nerf/helper.py:21

Functionmse2psnr

src/model/mipnerf/helper.py:22

Functionmse2psnr

src/model/plenoxel/utils.py:24

Functionmse2psnr

src/model/refnerf/helper.py:21

Functionmse2psnr

src/model/nerfpp/helper.py:20

Functionmse2psnr

src/model/aleth_nerf/helper.py:111

Functionmse2psnr

src/model/aleth_nerf_exp/helper.py:111

Functionnet_from_dict

(in_dict, prefix: str, model: nn.Module)

src/model/plenoxel/utils.py:539

Functionnet_to_dict

(out_dict: dict, prefix: str, model: nn.Module)

src/model/plenoxel/utils.py:527

utils/store_image.py:21

Methodon_after_backward

src/model/dvgo/model.py:471

Methodon_load_checkpoint

(self, checkpoint)

src/model/plenoxel/model.py:544

Methodon_load_checkpoint

(self, checkpoint: Dict[str, Any])

src/model/dvgo/model.py:824

Methodon_predict_epoch_end

(self, outputs)

src/model/interface.py:155

Methodon_save_checkpoint

(self, checkpoint)

src/model/plenoxel/model.py:504

Methodon_save_checkpoint

(self, checkpoint: Dict[str, Any])

src/model/dvgo/model.py:820

Methodon_train_batch_end

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

src/model/plenoxel/model.py:43

Methodon_train_batch_end

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

src/model/dvgo/model.py:111

Methodon_train_batch_start

(self, trainer, pl_module, batch, batch_idx)

src/model/dvgo/model.py:129

Methodoptim_basis_step

Execute RMSprop/SGD step on SH

src/model/plenoxel/sparse_grid.py:1206

Methodoptimizer_step

( self, epoch, batch_idx, optimizer, optimizer_idx, optimizer_

src/model/nerf/model.py:264

Methodoptimizer_step

( self, epoch, batch_idx, optimizer, optimizer_idx, optimizer_

src/model/mipnerf/model.py:271

Methodoptimizer_step

( self, epoch, batch_idx, optimizer, optimizer_idx, optimizer_

src/model/refnerf/model.py:388

Methodoptimizer_step

( self, epoch, batch_idx, optimizer, optimizer_idx, optimizer_

src/model/nerfpp/model.py:308

Methodoptimizer_step

( self, epoch, batch_idx, optimizer, optimizer_idx, optimizer_

src/model/aleth_nerf/model.py:365

Methodoptimizer_step

( self, epoch, batch_idx, optimizer, optimizer_idx, optimizer_

src/model/aleth_nerf_exp/model.py:358

Methodoptimizer_step

( self, epoch: int, batch_idx: int, optimizer, optimizer_idx,

src/model/dvgo/model.py:490

Methodoptimizer_zero_grad

(self, epoch, batch_idx, optimizer, optimizer_idx)

src/model/dvgo/model.py:404

Methodper_voxel_init

src/model/dvgo/model.py:691

Functionpos_enc

(x, min_deg, max_deg)

src/model/nerf/helper.py:59

Functionpos_enc

(x, min_deg, max_deg, append_identity)

src/model/mipnerf/helper.py:175

Functionpos_enc

(x, min_deg, max_deg, append_identity)

src/model/refnerf/helper.py:179

Functionpos_enc

(x, min_deg, max_deg)

src/model/nerfpp/helper.py:72

Functionpos_enc

(x, min_deg, max_deg)

src/model/aleth_nerf/helper.py:147

Functionpos_enc

(x, min_deg, max_deg)

src/model/aleth_nerf_exp/helper.py:147

Functionpose_interp

(poses, factor)

src/data/pose_utils.py:129

Methodpredict_dataloader

src/data/interface.py:293

Methodpredict_step

(self, *args, **kwargs)

src/model/interface.py:152

Functionptstocam

src/data/data_util/llff.py:167

Functionptstocam

src/data/data_util/nerf_360_v2.py:160

Functionptstocam

src/data/data_util/refnerf_real.py:158

Functionptstocam

src/data/data_util/llff_single.py:70

Functionptstocam

src/data/data_util/llff_pair.py:181

Functionraw2alpha

lib/dvgo/render_utils.cpp:120

Functionraw2alpha

lib/dvgo/cuda/render_utils.cpp:120

Functionraw2alpha_backward

lib/dvgo/render_utils.cpp:131

Functionraw2alpha_backward

lib/dvgo/cuda/render_utils.cpp:131

Functionraw2alpha_nonuni

lib/dvgo/render_utils.cpp:125

Functionraw2alpha_nonuni

lib/dvgo/cuda/render_utils.cpp:125

Functionraw2alpha_nonuni_backward

lib/dvgo/render_utils.cpp:136

Functionraw2alpha_nonuni_backward

lib/dvgo/cuda/render_utils.cpp:136

Functionrecenter_poses

src/data/data_util/nerf_360_v2.py:193

Functionrecenter_poses

src/data/data_util/refnerf_real.py:191

Functionreflect

Reflect view directions about normals. The reflection of a vector v about a unit vector n is a vector u such that dot(v, n) = dot(u, n), and

src/model/refnerf/ref_utils.py:16

Functionrender_path_spiral

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

src/data/data_util/nerf_360_v2.py:177

← previous next →401–500 of 605, ranked by callers