Functions — github.com/cmusatyalab/mega-nerf

hub / github.com/cmusatyalab/mega-nerf / functions

Functions102 in github.com/cmusatyalab/mega-nerf

⨍ Functions 102◇ Types & classes 11

↓ 30 callersMethodappend

(index: int)

mega_nerf/datasets/filesystem_dataset.py:317

↓ 25 callersFunctionmain_print

(log)

mega_nerf/misc_utils.py:6

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

scripts/colmap_to_mega_nerf.py:98

↓ 11 callersMethodeval

(self)

mega_nerf/runner.py:307

↓ 7 callersFunctionget_opts_base

()

mega_nerf/opts.py:4

↓ 5 callersFunctionget_ray_directions

(W: int, H: int, fx: float, fy: float, cx: float, cy: float, center_pixels: bool, devic

mega_nerf/ray_utils.py:6

↓ 4 callersFunction_depth2pts_outside

rays_o, rays_d: [..., 3] depth: [...]; inverse of distance to sphere origin

mega_nerf/rendering.py:420

↓ 4 callersFunction_get_single_nerf_inner

(hparams: Namespace, appearance_count: int, layer_dim: int, xyz_dim: int)

mega_nerf/models/model_utils.py:56

↓ 4 callersFunctionget_nerf

(hparams: Namespace, appearance_count: int)

mega_nerf/models/model_utils.py:12

↓ 4 callersFunctionget_rays

(directions: torch.Tensor, c2w: torch.Tensor, near: float, far: float, ray_altitude_range: List[f

mega_nerf/ray_utils.py:21

↓ 4 callersFunctionmain_tqdm

(inner)

mega_nerf/misc_utils.py:11

↓ 3 callersMethod_create_result_image

(rgbs: torch.Tensor, result_rgbs: torch.Tensor, result_depths: torch.Tensor)

mega_nerf/runner.py:592

↓ 3 callersFunction_expand_and_perturb_z_vals

(z_vals, samples, perturb, N_rays)

mega_nerf/rendering.py:472

↓ 3 callersFunction_get_results

(nerf: nn.Module, rays_d: torch.Tensor, image_indices: Optional[torch.Tensor

mega_nerf/rendering.py:176

↓ 3 callersMethod_run_validation

(self, train_index: int)

mega_nerf/runner.py:383

↓ 3 callersFunctionget_bg_nerf

(hparams: Namespace, appearance_count: int)

mega_nerf/models/model_utils.py:16

↓ 3 callersMethodget_state

(self)

mega_nerf/datasets/filesystem_dataset.py:79

↓ 3 callersMethodtrain

(self)

mega_nerf/runner.py:163

↓ 2 callersMethod__init__

(self, pos_xyz_dim: int, pos_dir_dim: int, layers: int, skip_layers: List[int], layer_dim: int,

mega_nerf/models/nerf.py:46

↓ 2 callersMethod_get_metadata_item

(self, metadata_path: Path, image_index: int, scale_factor: int, is_val: bool)

mega_nerf/runner.py:636

↓ 2 callersFunction_get_nerf_inner

(hparams: Namespace, appearance_count: int, layer_dim: int, xyz_dim: int, weight_key: str)

mega_nerf/models/model_utils.py:20

↓ 2 callersFunction_get_rays_inner

(rays_o: torch.Tensor, rays_d: torch.Tensor, near: float, far: float, ray_altitude_range:

mega_nerf/ray_utils.py:44

↓ 2 callersFunction_inference

(results: Dict[str, torch.Tensor], typ: str, nerf: nn.Module, ray

mega_nerf/rendering.py:251

↓ 2 callersFunction_render_images

(hparams: Namespace)

scripts/render_images.py:33

↓ 2 callersMethod_save_checkpoint

(self, optimizers: Dict[str, any], scaler: GradScaler, train_index: int, dataset_index: int,

mega_nerf/runner.py:519

↓ 2 callersMethod_setup_experiment_dir

(self)

mega_nerf/runner.py:324

↓ 2 callersFunction_truncate_with_plane_intersection

(rays_o: torch.Tensor, rays_d: torch.Tensor, altitude: float, default_bo

mega_nerf/ray_utils.py:65

↓ 2 callersMethod_write_final_metrics

(self, val_metrics: Dict[str, float])

mega_nerf/runner.py:312

↓ 2 callersMethod_write_to_disk

(self, executor: ThreadPoolExecutor, rgbs: torch.Tensor, rays: torch.FloatTensor, img_i

mega_nerf/datasets/filesystem_dataset.py:304

↓ 2 callersFunctiondetect_model_format

(path, ext)

scripts/colmap_to_mega_nerf.py:288

↓ 2 callersFunctionget_rgb_index_mask

(metadata: ImageMetadata)

mega_nerf/datasets/dataset_utils.py:8

↓ 2 callersMethodload_chunk

(self)

mega_nerf/datasets/filesystem_dataset.py:74

↓ 2 callersMethodload_image

(self)

mega_nerf/image_metadata.py:23

↓ 2 callersFunctionpsnr

(rgbs: torch.Tensor, target_rgbs: torch.Tensor)

mega_nerf/metrics.py:8

↓ 2 callersFunctionqvec2rotmat

(qvec)

scripts/colmap_to_mega_nerf.py:80

↓ 2 callersMethodrender_image

(self, metadata: ImageMetadata)

mega_nerf/runner.py:540

↓ 2 callersFunctionrender_rays

(nerf: nn.Module, bg_nerf: Optional[nn.Module], rays: torch.Tensor,

mega_nerf/rendering.py:15

↓ 2 callersMethodset_state

(self, chosen: str)

mega_nerf/datasets/filesystem_dataset.py:82

↓ 2 callersMethodvisualize_scalars

(scalar_tensor: torch.Tensor)

mega_nerf/runner.py:598

↓ 1 callersFunction_auto_scale

(hparams: Namespace, nerf: nn.Module, center: List[float], radius: List[float], device: torch.

scripts/create_octree.py:61

↓ 1 callersFunction_calculate_grid_weights

(hparams: Namespace, tree: N3Tree, poses: torch.Tensor, sigmas: torch.Tensor, reso

scripts/create_octree.py:108

↓ 1 callersMethod_check_existing_paths

(self, chunk_paths: List[Path], center_pixels: bool, scale_factor: int, images: int)

mega_nerf/datasets/filesystem_dataset.py:269

↓ 1 callersFunction_get_eval_opts

()

mega_nerf/eval.py:10

↓ 1 callersMethod_get_experiment_path

(self)

mega_nerf/runner.py:667

↓ 1 callersFunction_get_extraction_opts

()

scripts/create_octree.py:41

↓ 1 callersMethod_get_image_metadata

(self)

mega_nerf/runner.py:612

↓ 1 callersFunction_get_images_opts

()

scripts/copy_images.py:11

↓ 1 callersFunction_get_mask_opts

()

scripts/create_cluster_masks.py:19

↓ 1 callersFunction_get_merge_opts

()

scripts/convert_to_container.py:13

↓ 1 callersFunction_get_merge_opts

()

scripts/merge_submodules.py:13

↓ 1 callersFunction_get_opts

()

scripts/colmap_to_mega_nerf.py:323

↓ 1 callersFunction_get_render_opts

()

scripts/render_images.py:19

↓ 1 callersFunction_get_train_opts

()

mega_nerf/train.py:10

↓ 1 callersFunction_intersect_sphere

(rays_o: torch.Tensor, rays_d: torch.Tensor, sphere_center: torch.Tensor, sphere_radius:

mega_nerf/rendering.py:396

↓ 1 callersFunction_sample_cdf

(bins: torch.Tensor, cdf: torch.Tensor, fine_samples: int, det: bool)

mega_nerf/rendering.py:505

↓ 1 callersFunction_sample_pdf

Sample @N_importance samples from @bins with distribution defined by @weights. Inputs: bins: (N_rays, N_samples_+1) where N_samples_

mega_nerf/rendering.py:486

↓ 1 callersFunction_step1

(hparams: Namespace, nerf: nn.Module, tree: N3Tree, poses: torch.Tensor, device: torch.device)

scripts/create_octree.py:139

↓ 1 callersFunction_step2

(hparams: Namespace, nerf: nn.Module, tree: N3Tree, device: torch.device)

scripts/create_octree.py:189

↓ 1 callersMethod_training_step

(self, rgbs: torch.Tensor, rays: torch.Tensor, image_indices: Optional[torch.Tensor])

mega_nerf/runner.py:347

↓ 1 callersMethod_write_chunks

(self, metadata_items: List[ImageMetadata], center_pixels: bool, device: torch.device, c

mega_nerf/datasets/filesystem_dataset.py:134

↓ 1 callersFunctioneval_sh

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

mega_nerf/spherical_harmonics.py:55

↓ 1 callersFunctionget_rays_batch

(directions: torch.Tensor, c2w: torch.Tensor, near: float, far: float, ray_altitude_range:

mega_nerf/ray_utils.py:33

↓ 1 callersMethodload_mask

(self)

mega_nerf/image_metadata.py:32

↓ 1 callersFunctionlpips

(rgbs: torch.Tensor, target_rgbs: torch.Tensor)

mega_nerf/metrics.py:13

↓ 1 callersFunctionmain

(hparams: Namespace)

mega_nerf/eval.py:19

↓ 1 callersFunctionmain

(hparams: Namespace)

mega_nerf/train.py:19

↓ 1 callersFunctionmain

(hparams: Namespace)

scripts/copy_images.py:20

↓ 1 callersFunctionmain

(hparams: Namespace)

scripts/create_cluster_masks.py:36

↓ 1 callersFunctionmain

(hparams: Namespace)

scripts/convert_to_container.py:21

↓ 1 callersFunctionmain

(hparams: Namespace)

scripts/colmap_to_mega_nerf.py:336

↓ 1 callersFunctionmain

(hparams: Namespace)

scripts/merge_submodules.py:24

↓ 1 callersFunctionmain

(hparams: Namespace)

scripts/create_octree.py:213

↓ 1 callersFunctionmain

(hparams: Namespace)

scripts/render_images.py:133

↓ 1 callersFunctionread_cameras_binary

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

scripts/colmap_to_mega_nerf.py:175

↓ 1 callersFunctionread_cameras_text

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

scripts/colmap_to_mega_nerf.py:262

↓ 1 callersFunctionread_images_binary

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

scripts/colmap_to_mega_nerf.py:140

↓ 1 callersFunctionread_images_text

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

scripts/colmap_to_mega_nerf.py:231

↓ 1 callersFunctionread_model

(path, ext="")

scripts/colmap_to_mega_nerf.py:297

↓ 1 callersFunctionread_points3D_binary

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

scripts/colmap_to_mega_nerf.py:110

↓ 1 callersFunctionread_points3D_text

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

scripts/colmap_to_mega_nerf.py:204

↓ 1 callersFunctionssim

Computes SSIM from two images. This function was modeled after tf.image.ssim, and should produce comparable output. Args: rgbs: torc

mega_nerf/metrics.py:51

Method__getitem__

(self, idx)

mega_nerf/datasets/memory_dataset.py:58

Method__getitem__

(self, idx)

mega_nerf/datasets/filesystem_dataset.py:89

Method__init__

(self, hparams: Namespace, set_experiment_path: bool = True)

mega_nerf/runner.py:39

Method__init__

(self, image_path: Path, c2w: torch.Tensor, W: int, H: int, intrinsics: torch.Tensor, image_index: int,

mega_nerf/image_metadata.py:12

Method__init__

(self, metadata_items: List[ImageMetadata], near: float, far: float, ray_altitude_range: List[float],

mega_nerf/datasets/memory_dataset.py:14

Method__init__

(self, metadata_items: List[ImageMetadata], near: float, far: float, ray_altitude_range: List[float],

mega_nerf/datasets/filesystem_dataset.py:25

Method__init__

Defines a function that embeds x to (x, sin(2^k x), cos(2^k x), ...)

mega_nerf/models/nerf.py:9

Method__init__

(self, beta: int = 1, threshold: int = 20)

mega_nerf/models/nerf.py:33

Method__init__

(self, sub_modules: List[nn.Module], bg_sub_modules: List[nn.Module], centroids: torch.Tensor,

mega_nerf/models/mega_nerf_container.py:8

Method__init__

(self, coarse: nn.Module, fine: nn.Module)

mega_nerf/models/cascade.py:8

Method__init__

(self, sub_modules: List[nn.Module], centroids: torch.Tensor, boundary_margin: float, xyz_real: bool,

mega_nerf/models/mega_nerf.py:8

Method__len__

(self)

mega_nerf/datasets/memory_dataset.py:55

Method__len__

(self)

mega_nerf/datasets/filesystem_dataset.py:86

Method_load_chunk_inner

(self)

mega_nerf/datasets/filesystem_dataset.py:96

Methodextra_repr

(self)

mega_nerf/models/nerf.py:41

Methodforward

(self, x: torch.Tensor)

mega_nerf/models/nerf.py:20

Methodforward

(self, x: torch.Tensor)

mega_nerf/models/nerf.py:38

Methodforward

(self, x: torch.Tensor, sigma_only: bool = False, sigma_noise: Optional[torch.Tensor] = None)

mega_nerf/models/nerf.py:115

Methodforward

(self, use_coarse: bool, x: torch.Tensor, sigma_only: bool = False, sigma_noise: Optional[torc

mega_nerf/models/cascade.py:13

next →1–100 of 102, ranked by callers