SYMBOL INDEX (262 symbols across 18 files)

FILE: data/normalize_cameras.py
  function get_center_point (line 6) | def get_center_point(num_cams,cameras):
  function normalize_cameras (line 31) | def normalize_cameras(original_cameras_filename,output_cameras_filename,...

FILE: data/npz_to_blender.py
  function to16b (line 14) | def to16b(img):
  function opencv_to_gl (line 19) | def opencv_to_gl(pose):
  function get_offset (line 25) | def get_offset(poses):
  function scale_pose (line 37) | def scale_pose(pose, scale, offset):
  function load_K_Rt_from_P (line 43) | def load_K_Rt_from_P(filename, P=None):
  function main (line 67) | def main():

FILE: dataset/eval_dataset.py
  class GridDataset (line 15) | class GridDataset(Dataset):
    method __init__ (line 19) | def __init__(self, points, xyz) -> None:
    method __len__ (line 24) | def __len__(self):
    method __getitem__ (line 27) | def __getitem__(self, index):
  class PlotDataset (line 31) | class PlotDataset(torch.utils.data.Dataset):
    method __init__ (line 32) | def __init__(self,
    method shuffle_plot_index (line 137) | def shuffle_plot_index(self):
    method __len__ (line 141) | def __len__(self):
    method get_uv (line 144) | def get_uv(self):
    method __getitem__ (line 150) | def __getitem__(self, idx):
    method collate_fn (line 170) | def collate_fn(self, batch_list):
  class InterpolateDataset (line 188) | class InterpolateDataset(torch.utils.data.Dataset):
    method __init__ (line 192) | def __init__(self,
    method __len__ (line 244) | def __len__(self):
    method __getitem__ (line 247) | def __getitem__(self, idx):
    method collate_fn (line 258) | def collate_fn(self, batch_list):
  class RelightDataset (line 276) | class RelightDataset(PlotDataset):
    method __init__ (line 277) | def __init__(self, data_dir, edit_cfg, scan_id=0, is_val=False, **kwar...
    method loadattr (line 302) | def loadattr(self, edit_cfg, attr, mode=0):
    method __len__ (line 316) | def __len__(self):
    method __getitem__ (line 319) | def __getitem__(self, idx):
  class RelightVideoDataset (line 343) | class RelightVideoDataset(PlotDataset):
    method __init__ (line 344) | def __init__(self, data_dir, edit_cfg, scan_id=0, is_val=False, **kwar...
    method __len__ (line 358) | def __len__(self):
    method __getitem__ (line 361) | def __getitem__(self, idx):

FILE: dataset/train_dataset.py
  class ReconDataset (line 15) | class ReconDataset(torch.utils.data.Dataset):
    method __init__ (line 17) | def __init__(self,
    method __len__ (line 166) | def __len__(self):
    method __getitem__ (line 169) | def __getitem__(self, idx):
    method collate_fn (line 194) | def collate_fn(self, batch_list):
  class MaterialDataset (line 212) | class MaterialDataset(torch.utils.data.Dataset):
    method __init__ (line 214) | def __init__(self,
    method __len__ (line 315) | def __len__(self):
    method __getitem__ (line 318) | def __getitem__(self, idx):
    method collate_fn (line 336) | def collate_fn(self, batch_list):

FILE: model/eval/recon.py
  class SDFMeshSystem (line 21) | class SDFMeshSystem(pl.LightningModule):
    method __init__ (line 22) | def __init__(self, conf, exp_dir, resolution, score=False, far_clip=5....
    method initialize (line 46) | def initialize(self):
    method test_dataloader (line 84) | def test_dataloader(self):
    method test_step (line 89) | def test_step(self, batch, batch_idx):
    method test_epoch_end (line 92) | def test_epoch_end(self, outputs) -> None:
    method forward (line 131) | def forward(self):
  class VolumeRenderSystem (line 135) | class VolumeRenderSystem(pl.LightningModule):
    method __init__ (line 136) | def __init__(self, conf, exp_dir, indices=None, is_val=False, score_me...
    method test_dataloader (line 157) | def test_dataloader(self):
    method test_step (line 163) | def test_step(self, batch, batch_idx):
    method test_epoch_end (line 205) | def test_epoch_end(self, outputs):
    method forward (line 223) | def forward(self):
  class ViewInterpolateSystem (line 227) | class ViewInterpolateSystem(pl.LightningModule):
    method __init__ (line 228) | def __init__(self, conf, exp_dir, id0, id1, n_frames=60, frame_rate=24...
    method test_dataloader (line 253) | def test_dataloader(self):
    method test_step (line 259) | def test_step(self, batch, batch_idx):
    method test_epoch_end (line 284) | def test_epoch_end(self, outputs):
    method forward (line 302) | def forward(self):

FILE: model/network/__init__.py
  class I2SDFNetwork (line 19) | class I2SDFNetwork(nn.Module):
    method __init__ (line 20) | def __init__(self, conf):
    method init_emission_groups (line 49) | def init_emission_groups(self, n_emitters, pointcloud, init_emission=1...
    method get_param_groups (line 77) | def get_param_groups(self, lr):
    method forward (line 80) | def forward(self, input, predict_only=False):
    method volume_rendering (line 223) | def volume_rendering(self, z_vals, z_max, sdf):
    method bg_volume_rendering (line 242) | def bg_volume_rendering(self, z_vals_bg, bg_sdf):
    method depth2pts_outside (line 258) | def depth2pts_outside(self, ray_o, ray_d, depth):
  class I2SDFLoss (line 289) | class I2SDFLoss(nn.Module):
    method __init__ (line 290) | def __init__(self, eikonal_weight=0.1, smooth_weight=0.0, mask_weight=...
    method get_rgb_loss (line 308) | def get_rgb_loss(self, rgb_values, rgb_gt):
    method get_eikonal_loss (line 313) | def get_eikonal_loss(self, grad_theta):
    method get_mask_loss (line 317) | def get_mask_loss(self, mask_pred, mask_gt):
    method get_depth_loss (line 320) | def get_depth_loss(self, depth, depth_gt, depth_mask):
    method get_normal_l1_loss (line 326) | def get_normal_l1_loss(self, normal, normal_gt, normal_mask):
    method get_normal_angular_loss (line 331) | def get_normal_angular_loss(self, normal, normal_gt, normal_mask):
    method forward (line 338) | def forward(self, model_outputs, ground_truth, current_step):

FILE: model/network/density.py
  class Density (line 5) | class Density(nn.Module):
    method __init__ (line 6) | def __init__(self, params_init={}):
    method forward (line 12) | def forward(self, sdf, beta=None):
  class LaplaceDensity (line 16) | class LaplaceDensity(Density):  # alpha * Laplace(loc=0, scale=beta).cdf...
    method __init__ (line 17) | def __init__(self, params_init={}, beta_min=0.0001):
    method density_func (line 21) | def density_func(self, sdf, beta=None):
    method get_beta (line 28) | def get_beta(self):
  class AbsDensity (line 33) | class AbsDensity(Density):  # like NeRF++
    method density_func (line 34) | def density_func(self, sdf, beta=None):
  class SimpleDensity (line 38) | class SimpleDensity(Density):  # like NeRF
    method __init__ (line 39) | def __init__(self, params_init={}, noise_std=1.0):
    method density_func (line 43) | def density_func(self, sdf, beta=None):

FILE: model/network/embedder.py
  class Embedder (line 6) | class Embedder:
    method __init__ (line 8) | def __init__(self, **kwargs):
    method create_embedding_fn (line 12) | def create_embedding_fn(self):
    method embed (line 37) | def embed(self, inputs):
  class SHEncoder (line 41) | class SHEncoder(nn.Module):
    method __init__ (line 42) | def __init__(self, input_dims=3, degree=4):
    method forward (line 84) | def forward(self, input, **kwargs):
  class FourierFeature (line 125) | class FourierFeature(nn.Module):
    method __init__ (line 126) | def __init__(self, channels, sigma=1.0, input_dims=3, include_input=Tr...
    method forward (line 133) | def forward(self, x):
  function get_embedder (line 138) | def get_embedder(embed_type='positional', **kwargs):

FILE: model/network/mlp.py
  class ImplicitNetwork (line 10) | class ImplicitNetwork(nn.Module):
    method __init__ (line 11) | def __init__(
    method get_param_groups (line 81) | def get_param_groups(self, lr):
    method forward (line 84) | def forward(self, input):
    method gradient (line 107) | def gradient(self, x):
    method feature (line 120) | def feature(self, x):
    method get_outputs (line 123) | def get_outputs(self, x, returns_grad=True):
    method get_sdf_vals (line 145) | def get_sdf_vals(self, x):
  class RenderingNetwork (line 159) | class RenderingNetwork(nn.Module):
    method __init__ (line 160) | def __init__(
    method forward (line 208) | def forward(self, points, normals, view_dirs, feature_vectors):

FILE: model/network/ray_sampler.py
  class RaySampler (line 6) | class RaySampler(metaclass=abc.ABCMeta):
    method __init__ (line 7) | def __init__(self, near, far):
    method get_z_vals (line 12) | def get_z_vals(self, ray_dirs, cam_loc, model):
  class UniformSampler (line 15) | class UniformSampler(RaySampler):
    method __init__ (line 16) | def __init__(self, scene_bounding_sphere, near, N_samples, take_sphere...
    method get_z_vals (line 22) | def get_z_vals(self, ray_dirs, cam_loc, model):
  class ErrorBoundSampler (line 46) | class ErrorBoundSampler(RaySampler):
    method __init__ (line 47) | def __init__(self, scene_bounding_sphere, near, N_samples, N_samples_e...
    method get_z_vals (line 67) | def get_z_vals(self, ray_dirs, cam_loc, model):
    method get_error_bound (line 243) | def get_error_bound(self, beta, model, sdf, z_vals, dists, d_star):

FILE: model/rendering/__init__.py
  class RenderingLayer (line 10) | class RenderingLayer(nn.Module):
    method __init__ (line 11) | def __init__(self, spp, split_n_pixels, preserve_light=True) -> None:
    method forward (line 17) | def forward(

FILE: model/rendering/brdf.py
  function create_frame (line 5) | def create_frame(n: torch.Tensor, eps:float = 1e-6):
  function get_rendering_parameters (line 20) | def get_rendering_parameters(albedo_raw, rough_raw, use_metallic):
  function to_global (line 35) | def to_global(d, x, y, z):
  function sqrt_ (line 41) | def sqrt_(x: torch.Tensor, eps=1e-8) -> torch.Tensor:
  function reflect (line 47) | def reflect(v: torch.Tensor, h: torch.Tensor):
  function square_to_cosine_hemisphere (line 51) | def square_to_cosine_hemisphere(sample: torch.Tensor):
  function get_cos_theta (line 59) | def get_cos_theta(v: torch.Tensor):
  function get_phi (line 63) | def get_phi(v: torch.Tensor):
  function sample_disney_specular (line 72) | def sample_disney_specular(sample: torch.Tensor, roughness: torch.Tensor...
  function GTR2 (line 94) | def GTR2(ndh, a):
  function SchlickFresnel (line 99) | def SchlickFresnel(u):
  function smithG_GGX (line 103) | def smithG_GGX(ndv, a):
  function pdf_disney (line 109) | def pdf_disney(roughness: torch.Tensor, metallic: torch.Tensor, wi: torc...
  function eval_disney (line 130) | def eval_disney(albedo: torch.Tensor, roughness: torch.Tensor, metallic:...
  function F_Schlick (line 164) | def F_Schlick(SpecularColor, VoH):
  function GetSpecularEventProbability (line 168) | def GetSpecularEventProbability(SpecularColor, NoV) -> torch.Tensor:
  function baseColorToSpecularF0 (line 172) | def baseColorToSpecularF0(baseColor, metalness):
  function luminance (line 175) | def luminance(color):
  function probabilityToSampleSpecular (line 181) | def probabilityToSampleSpecular(difColor, specColor) -> torch.Tensor:
  function shadowedF90 (line 186) | def shadowedF90(F0):
  function evalFresnel (line 190) | def evalFresnel(f0, f90, NdotS):
  function Smith_G1_GGX (line 194) | def Smith_G1_GGX(alphaSquared, NdotSSquared):
  function Smith_G2_GGX (line 197) | def Smith_G2_GGX(alphaSquared, NdotL, NdotV):
  function GGX_D (line 202) | def GGX_D(alphaSquared, NdotH):
  function pdf_ggx (line 206) | def pdf_ggx(Kd: torch.Tensor, Ks: torch.Tensor, roughness: torch.Tensor,...
  function eval_ggx (line 241) | def eval_ggx(Kd: torch.Tensor, Ks: torch.Tensor, roughness: torch.Tensor...
  function sample_weight_ggx (line 268) | def sample_weight_ggx(alphaSquared, NdotL, NdotV):
  function sample_ggx (line 273) | def sample_ggx(sample: torch.Tensor, Kd: torch.Tensor, Ks: torch.Tensor,...
  function sample_ggx_specular (line 325) | def sample_ggx_specular(sample: torch.Tensor, roughness: torch.Tensor, w...

FILE: model/trainer/recon.py
  class ReconstructionTrainer (line 23) | class ReconstructionTrainer(pl.LightningModule):
    method __init__ (line 24) | def __init__(self, conf, prog_bar: RichProgressBar, exp_dir, model_onl...
    method forward (line 109) | def forward(self):
    method plot_hotmap (line 112) | def plot_hotmap(self, path):
    method plot_countmap (line 127) | def plot_countmap(self, path):
    method update_pdf (line 142) | def update_pdf(self, value, idx):
    method sample_bubble (line 155) | def sample_bubble(self, batch_size):
    method initialize_bubble_pdf (line 172) | def initialize_bubble_pdf(self, split_size):
    method configure_optimizers (line 201) | def configure_optimizers(self):
    method train_dataloader (line 209) | def train_dataloader(self):
    method val_dataloader (line 212) | def val_dataloader(self):
    method log_if_nonzero (line 215) | def log_if_nonzero(self, name, value, *args, **kwargs):
    method training_step (line 219) | def training_step(self, batch, batch_idx):
    method validation_step (line 290) | def validation_step(self, batch, batch_idx):
    method validation_epoch_end (line 358) | def validation_epoch_end(self, outputs) -> None:

FILE: utils/__init__.py
  class RichProgressBarWithScanId (line 12) | class RichProgressBarWithScanId(RichProgressBar):
    method __init__ (line 13) | def __init__(self, scan_id, *args, **kwargs) -> None:
    method configure_columns (line 17) | def configure_columns(self, trainer):
  function glob_imgs (line 21) | def glob_imgs(path):
  function glob_depths (line 27) | def glob_depths(path):
  function split_input (line 35) | def split_input(model_input, total_pixels, n_pixels=10000):
  function split_dict (line 50) | def split_dict(d, batch_size=10000):
  function detach_dict (line 66) | def detach_dict(d):
  function merge_output (line 70) | def merge_output(res, total_pixels, batch_size):
  function merge_dict (line 87) | def merge_dict(dicts):
  class _trunc_exp (line 96) | class _trunc_exp(Function):
    method forward (line 99) | def forward(ctx, x):
    method backward (line 105) | def backward(ctx, g):
  function kmeans_pp_centroid (line 111) | def kmeans_pp_centroid(points: torch.Tensor, k):

FILE: utils/cfgnode.py
  class CfgNode (line 34) | class CfgNode(dict):
    method __init__ (line 44) | def __init__(
    method _create_config_tree_from_dict (line 87) | def _create_config_tree_from_dict(cls, init_dict: dict, key_list: list):
    method __getattr__ (line 110) | def __getattr__(self, name: str):
    method __setattr__ (line 116) | def __setattr__(self, name: str, value):
    method __str__ (line 138) | def __str__(self):
    method __repr__ (line 159) | def __repr__(self):
    method dump (line 162) | def dump(self, **kwargs):
    method merge_from_file (line 184) | def merge_from_file(self, cfg_filename: str):
    method merge_from_other_cfg (line 193) | def merge_from_other_cfg(self, cfg_other):
    method merge_from_list (line 200) | def merge_from_list(self, cfg_list: list):
    method freeze (line 229) | def freeze(self):
    method defrost (line 233) | def defrost(self):
    method is_frozen (line 237) | def is_frozen(self):
    method _immutable (line 241) | def _immutable(self, is_immutable: bool):
    method clone (line 255) | def clone(self):
    method register_deprecated_key (line 259) | def register_deprecated_key(self, key: str):
    method register_renamed_key (line 270) | def register_renamed_key(
    method key_is_deprecated (line 286) | def key_is_deprecated(self, full_key: str):
    method key_is_renamed (line 293) | def key_is_renamed(self, full_key: str):
    method raise_key_rename_error (line 297) | def raise_key_rename_error(self, full_key: str):
    method is_new_allowed (line 310) | def is_new_allowed(self):
    method load_cfg (line 314) | def load_cfg(cls, cfg_file_obj_or_str):
    method _load_cfg_from_file (line 336) | def _load_cfg_from_file(cls, file_obj):
    method _load_cfg_from_yaml_str (line 351) | def _load_cfg_from_yaml_str(cls, str_obj):
    method _load_cfg_py_source (line 357) | def _load_cfg_py_source(cls, filename):
    method _decode_cfg_value (line 374) | def _decode_cfg_value(cls, value):
  function _valid_type (line 406) | def _valid_type(value, allow_cfg_node: Optional[bool] = False):
  function _merge_a_into_b (line 412) | def _merge_a_into_b(a: CfgNode, b: CfgNode, root: CfgNode, key_list: list):
  function _check_and_coerce_cfg_value_type (line 450) | def _check_and_coerce_cfg_value_type(replacement, original, key, full_key):
  function _assert_with_logging (line 482) | def _assert_with_logging(cond, msg):
  function _load_module_from_file (line 488) | def _load_module_from_file(name, filename):

FILE: utils/mesh_util.py
  function nn_correspondance (line 12) | def nn_correspondance(verts1, verts2):
  function evaluate (line 25) | def evaluate(mesh_pred, mesh_trgt, threshold=.05, down_sample=.02):
  class Renderer (line 55) | class Renderer():
    method __init__ (line 56) | def __init__(self, height=480, width=640):
    method __call__ (line 61) | def __call__(self, height, width, intrinsics, pose, mesh):
    method fix_pose (line 71) | def fix_pose(self, pose):
    method mesh_opengl (line 83) | def mesh_opengl(self, mesh):
    method delete (line 86) | def delete(self):
  function refuse (line 90) | def refuse(mesh, poses, K, H, W, far_clip=5.0):
  function depth2mesh (line 117) | def depth2mesh(depths, poses, K, H, W):

FILE: utils/plots.py
  function plot (line 15) | def plot(implicit_network, indices, plot_data, path, epoch, img_res, plo...
  function visualize_pointcloud (line 76) | def visualize_pointcloud(points, filename):
  function visualize_clustered_pointcloud (line 90) | def visualize_clustered_pointcloud(points, labels, centroids, filename):
  function visualize_marked_pointcloud (line 105) | def visualize_marked_pointcloud(points, counts, path, epoch):
  function get_3D_scatter_trace (line 120) | def get_3D_scatter_trace(points, name='', size=3, caption=None):
  function get_3D_marked_scatter_trace (line 141) | def get_3D_marked_scatter_trace(points, marks, name='', size=1, caption=...
  function get_3D_quiver_trace (line 164) | def get_3D_quiver_trace(points, directions, color='#bd1540', name=''):
  function get_surface_trace (line 188) | def get_surface_trace(path, epoch, sdf, resolution=100, grid_boundary=[-...
  function extract_fields (line 228) | def extract_fields(bound_min, bound_max, resolution, query_func):
  function extract_geometry (line 246) | def extract_geometry(bound_min, bound_max, resolution, threshold, query_...
  function get_surface_high_res_mesh (line 258) | def get_surface_high_res_mesh(sdf, resolution=100, grid_boundary=[-2.0, ...
  function get_surface_by_grid (line 339) | def get_surface_by_grid(grid_params, sdf, resolution=100, level=0, highe...
  function get_grid_uniform (line 440) | def get_grid_uniform(resolution, grid_boundary=[-2.0, 2.0]):
  function get_grid (line 453) | def get_grid(points, resolution, input_min=None, input_max=None, eps=0.1):
  function plot_imgs_wo_gt (line 492) | def plot_imgs_wo_gt(normal_maps, path, epoch, plot_nrow, img_res, path_n...
  function plot_imgs_filter (line 508) | def plot_imgs_filter(rgb_points, ground_true, path, epoch, img_res, path...
  function plot_colormap (line 522) | def plot_colormap(mat_info, path, epoch, plot_nrow, img_res, colormap=cv...
  function plot_depths (line 538) | def plot_depths(depth_maps, path, epoch, plot_nrow, img_res, colormap=cv...
  function plot_images (line 560) | def plot_images(rgb_points, ground_true, path, epoch, plot_nrow, img_res...
  function lin2img (line 581) | def lin2img(tensor, img_res):

FILE: utils/rend_util.py
  function linear_to_srgb (line 9) | def linear_to_srgb(data):
  function get_psnr (line 13) | def get_psnr(img1, img2, normalize_rgb=False):
  function load_rgb (line 25) | def load_rgb(path, normalize_rgb = False, is_hdr = False):
  function load_mask (line 38) | def load_mask(path):
  function load_depth (line 46) | def load_depth(path):
  function load_normal (line 52) | def load_normal(path):
  function load_K_Rt_from_P (line 57) | def load_K_Rt_from_P(filename, P=None):
  function depth_to_world (line 81) | def depth_to_world(uv, intrinsics, pose, depth, depth_mask=None):
  function get_camera_params (line 92) | def get_camera_params(uv, pose, intrinsics):
  function get_camera_for_plot (line 123) | def get_camera_for_plot(pose):
  function lift (line 134) | def lift(x, y, z, intrinsics):
  function quat_to_rot (line 150) | def quat_to_rot(q):
  function rot_to_quat (line 170) | def rot_to_quat(R):
  function get_general_sphere_intersections (line 191) | def get_general_sphere_intersections(cam_loc, ray_directions, center, r):
  function get_sphere_intersections (line 211) | def get_sphere_intersections(cam_loc, ray_directions, r = 1.0):
  function add_depth_noise (line 229) | def add_depth_noise(depth, depth_mask, scale=1):