SYMBOL INDEX (257 symbols across 25 files)

FILE: 2D_regression.py
  function PSNR (line 17) | def PSNR(a, b):
  function rgb_ssim (line 26) | def rgb_ssim(img0, img1, max_val,
  function eval_img (line 76) | def eval_img(aabb, reso, shiftment=[0.5, 0.5], chunk=10240):
  function linear_to_srgb (line 95) | def linear_to_srgb(img):
  function write_image_imageio (line 100) | def write_image_imageio(img_file, img, colormap=None, quality=100):

FILE: dataLoader/blender.py
  class BlenderDataset (line 12) | class BlenderDataset(Dataset):
    method __init__ (line 13) | def __init__(self, cfg, split='train', batch_size=4096, is_stack=None):
    method read_depth (line 40) | def read_depth(self, filename):
    method read_meta (line 44) | def read_meta(self):
    method define_transforms (line 105) | def define_transforms(self):
    method define_proj_mat (line 108) | def define_proj_mat(self):
    method __len__ (line 115) | def __len__(self):
    method __getitem__ (line 118) | def __getitem__(self, idx):

FILE: dataLoader/blender_set.py
  class BlenderDatasetSet (line 12) | class BlenderDatasetSet(Dataset):
    method __init__ (line 13) | def __init__(self, cfg, split='train'):
    method read_depth (line 39) | def read_depth(self, filename):
    method read_meta (line 43) | def read_meta(self):
    method define_transforms (line 103) | def define_transforms(self):
    method define_proj_mat (line 106) | def define_proj_mat(self):
    method __len__ (line 113) | def __len__(self):
    method __getitem__ (line 116) | def __getitem__(self, idx):

FILE: dataLoader/colmap.py
  class ColmapDataset (line 12) | class ColmapDataset(Dataset):
    method __init__ (line 13) | def __init__(self, cfg, split='train'):
    method read_meta (line 31) | def read_meta(self):
    method define_transforms (line 107) | def define_transforms(self):
    method __len__ (line 111) | def __len__(self):
    method __getitem__ (line 114) | def __getitem__(self, idx):

FILE: dataLoader/colmap2nerf.py
  function parse_args (line 23) | def parse_args():
  function do_system (line 44) | def do_system(arg):
  function run_ffmpeg (line 51) | def run_ffmpeg(args):
  function run_colmap (line 73) | def run_colmap(args):
  function variance_of_laplacian (line 107) | def variance_of_laplacian(image):
  function sharpness (line 110) | def sharpness(imagePath):
  function qvec2rotmat (line 116) | def qvec2rotmat(qvec):
  function rotmat (line 133) | def rotmat(a, b):
  function closest_point_2_lines (line 141) | def closest_point_2_lines(oa, da, ob, db): # returns point closest to bo...
  function normalize (line 156) | def normalize(x):
  function rotation_matrix_from_vectors (line 159) | def rotation_matrix_from_vectors(vec1, vec2):
  function rotation_up (line 176) | def rotation_up(poses):
  function search_orientation (line 181) | def search_orientation(points):
  function load_point_txt (line 196) | def load_point_txt(path):

FILE: dataLoader/dtu_objs.py
  function load_K_Rt_from_P (line 11) | def load_K_Rt_from_P(filename, P=None):
  function fps_downsample (line 34) | def fps_downsample(points, n_points_to_sample):
  class DTUDataset (line 47) | class DTUDataset(Dataset):
    method __init__ (line 48) | def __init__(self, cfg, split='train', batch_size=4096, is_stack=None):
    method read_meta (line 78) | def read_meta(self):
    method load_data (line 122) | def load_data(self, scene_idx, img_idx=None):
    method get_bbox (line 196) | def get_bbox(self, scale_mats_np, object_scale_mat):
    method gen_rays_at (line 206) | def gen_rays_at(self, intrinsic, c2w, resolution_level=1):
    method __len__ (line 224) | def __len__(self):
    method __getitem__ (line 227) | def __getitem__(self, idx):

FILE: dataLoader/dtu_objs2.py
  function load_K_Rt_from_P (line 12) | def load_K_Rt_from_P(filename, P=None):
  class DTUDataset (line 35) | class DTUDataset(Dataset):
    method __init__ (line 36) | def __init__(self, cfg, split='train', batch_size=4096, is_stack=None):
    method get_bbox (line 57) | def get_bbox(self):
    method gen_rays_at (line 68) | def gen_rays_at(self, intrinsic, c2w, resolution_level=1):
    method read_meta (line 85) | def read_meta(self):
    method __len__ (line 131) | def __len__(self):
    method __getitem__ (line 134) | def __getitem__(self, idx):

FILE: dataLoader/google_objs.py
  function fps_downsample (line 13) | def fps_downsample(points, n_points_to_sample):
  function pose_spherical_nerf (line 29) | def pose_spherical_nerf(euler, radius=1.8, ep=1):
  function nerf_video_path (line 36) | def nerf_video_path(c2ws, theta_range=10,phi_range=20,N_views=120,radius...
  function _interpolate_trajectory (line 56) | def _interpolate_trajectory(c2ws, num_views: int = 300):
  function google_objs_path (line 77) | def google_objs_path(c2ws, N_views=150):
  class GoogleObjsDataset (line 85) | class GoogleObjsDataset(Dataset):
    method __init__ (line 86) | def __init__(self, cfg, split="train", batch_size=4096):
    method read_depth (line 174) | def read_depth(self, filename):
    method read_meta (line 178) | def read_meta(self):
    method get_rays (line 301) | def get_rays(self, idx):
    method get_rgbs (line 306) | def get_rgbs(self, idx):
    method define_transforms (line 312) | def define_transforms(self):
    method define_proj_mat (line 315) | def define_proj_mat(self):
    method world2ndc (line 320) | def world2ndc(self, points, lindisp=None):
    method update_index (line 324) | def update_index(self):
    method __len__ (line 327) | def __len__(self):
    method __getitem__ (line 330) | def __getitem__(self, idx):

FILE: dataLoader/image.py
  function load (line 10) | def load(path):
  function srgb_to_linear (line 22) | def srgb_to_linear(img):
  class ImageDataset (line 26) | class ImageDataset(Dataset):
    method __init__ (line 27) | def __init__(self, cfg, batchsize, split='train', continue_sampling=Fa...
    method __len__ (line 84) | def __len__(self):
    method __getitem__ (line 87) | def __getitem__(self, idx):

FILE: dataLoader/image_set.py
  function srgb_to_linear (line 6) | def srgb_to_linear(img):
  function load (line 10) | def load(path, HW=512):
  class ImageSetDataset (line 24) | class ImageSetDataset(Dataset):
    method __init__ (line 25) | def __init__(self, cfg, batchsize, split='train', continue_sampling=Fa...
    method __len__ (line 50) | def __len__(self):
    method __getitem__ (line 53) | def __getitem__(self, idx):

FILE: dataLoader/llff.py
  function normalize (line 12) | def normalize(v):
  function average_poses (line 17) | def average_poses(poses):
  function center_poses (line 54) | def center_poses(poses, blender2opencv):
  function viewmatrix (line 81) | def viewmatrix(z, up, pos):
  function render_path_spiral (line 91) | def render_path_spiral(c2w, up, rads, focal, zdelta, zrate, N_rots=2, N=...
  function get_spiral (line 102) | def get_spiral(c2ws_all, near_fars, rads_scale=1.0, N_views=120):
  class LLFFDataset (line 122) | class LLFFDataset(Dataset):
    method __init__ (line 123) | def __init__(self, cfg , split='train', hold_every=8):
    method read_meta (line 148) | def read_meta(self):
    method define_transforms (line 231) | def define_transforms(self):
    method __len__ (line 234) | def __len__(self):
    method __getitem__ (line 237) | def __getitem__(self, idx):

FILE: dataLoader/nsvf.py
  function pose_spherical (line 29) | def pose_spherical(theta, phi, radius):
  class NSVF (line 36) | class NSVF(Dataset):
    method __init__ (line 38) | def __init__(self, datadir, split='train', downsample=1.0, wh=[800,800...
    method bbox2corners (line 56) | def bbox2corners(self):
    method read_meta (line 63) | def read_meta(self):
    method define_transforms (line 132) | def define_transforms(self):
    method define_proj_mat (line 135) | def define_proj_mat(self):
    method world2ndc (line 138) | def world2ndc(self, points):
    method __len__ (line 142) | def __len__(self):
    method __getitem__ (line 147) | def __getitem__(self, idx):

FILE: dataLoader/ray_utils.py
  function load_json (line 7) | def load_json(path):
  function depth2dist (line 12) | def depth2dist(z_vals, cos_angle):
  function ndc2dist (line 21) | def ndc2dist(ndc_pts, cos_angle):
  function get_ray_directions (line 27) | def get_ray_directions(H, W, focal, center=None):
  function get_ray_directions_blender (line 48) | def get_ray_directions_blender(H, W, focal, center=None):
  function get_rays (line 69) | def get_rays(directions, c2w):
  function ndc_rays_blender (line 93) | def ndc_rays_blender(H, W, focal, near, rays_o, rays_d):
  function ndc_rays (line 112) | def ndc_rays(H, W, focal, near, rays_o, rays_d):
  function sample_pdf (line 132) | def sample_pdf(bins, weights, N_samples, det=False, pytest=False):
  function dda (line 177) | def dda(rays_o, rays_d, bbox_3D):
  function ray_marcher (line 187) | def ray_marcher(rays,
  function read_pfm (line 234) | def read_pfm(filename):
  class SimpleSampler (line 271) | class SimpleSampler:
    method __init__ (line 272) | def __init__(self, total, batch):
    method nextids (line 278) | def nextids(self):
  function ndc_bbox (line 285) | def ndc_bbox(all_rays):
  function pose_from_json (line 293) | def pose_from_json(meta, transpose):
  function rotation_matrix_from_vectors (line 300) | def rotation_matrix_from_vectors(vec1, vec2):
  function normalize (line 317) | def normalize(x):
  function rotation_up (line 320) | def rotation_up(poses):
  function search_orientation (line 325) | def search_orientation(points):
  function load_point_txt (line 341) | def load_point_txt(path):
  function orientation (line 365) | def orientation(poses, point_path=None):
  function spherify_poses (line 397) | def spherify_poses(poses, radus=1):

FILE: dataLoader/sdf.py
  function N_to_reso (line 5) | def N_to_reso(avg_reso, bbox):
  function load (line 12) | def load(path, split, dtype='points'):
  class SDFDataset (line 31) | class SDFDataset(Dataset):
    method __init__ (line 32) | def __init__(self, cfg, split='train'):
    method __len__ (line 42) | def __len__(self):
    method __getitem__ (line 45) | def __getitem__(self, idx):

FILE: dataLoader/tankstemple.py
  function circle (line 11) | def circle(radius=3.5, h=0.0, axis='z', t0=0, r=1):
  function cross (line 20) | def cross(x, y, axis=0):
  function normalize (line 25) | def normalize(x, axis=-1, order=2):
  function cat (line 37) | def cat(x, axis=1):
  function look_at_rotation (line 43) | def look_at_rotation(camera_position, at=None, up=None, inverse=False, c...
  function gen_path (line 76) | def gen_path(pos_gen, at=(0, 0, 0), up=(0, -1, 0), frames=180):
  class TanksTempleDataset (line 87) | class TanksTempleDataset(Dataset):
    method __init__ (line 90) | def __init__(self, cfg, split='train'):
    method bbox2corners (line 109) | def bbox2corners(self):
    method read_meta (line 115) | def read_meta(self):
    method define_transforms (line 192) | def define_transforms(self):
    method define_proj_mat (line 195) | def define_proj_mat(self):
    method world2ndc (line 198) | def world2ndc(self, points):
    method __len__ (line 202) | def __len__(self):
    method __getitem__ (line 207) | def __getitem__(self, idx):

FILE: dataLoader/your_own_data.py
  class YourOwnDataset (line 13) | class YourOwnDataset(Dataset):
    method __init__ (line 14) | def __init__(self, datadir, split='train', downsample=1.0, is_stack=Fa...
    method read_depth (line 35) | def read_depth(self, filename):
    method read_meta (line 39) | def read_meta(self):
    method define_transforms (line 102) | def define_transforms(self):
    method define_proj_mat (line 105) | def define_proj_mat(self):
    method world2ndc (line 108) | def world2ndc(self,points,lindisp=None):
    method __len__ (line 112) | def __len__(self):
    method __getitem__ (line 115) | def __getitem__(self, idx):

FILE: models/FactorFields.py
  function grid_mapping (line 11) | def grid_mapping(positions, freq_bands, aabb, basis_mapping='sawtooth'):
  function dct_dict (line 36) | def dct_dict(n_atoms_fre, size, n_selete, dim=2):
  function positional_encoding (line 74) | def positional_encoding(positions, freqs):
  function raw2alpha (line 82) | def raw2alpha(sigma, dist):
  class AlphaGridMask (line 91) | class AlphaGridMask(torch.nn.Module):
    method __init__ (line 92) | def __init__(self, device, aabb, alpha_volume):
    method sample_alpha (line 103) | def sample_alpha(self, xyz_sampled):
    method normalize_coord (line 109) | def normalize_coord(self, xyz_sampled):
  class MLPMixer (line 113) | class MLPMixer(torch.nn.Module):
    method __init__ (line 114) | def __init__(self,
    method forward (line 144) | def forward(self, x, is_train=False):
  class MLPRender_Fea (line 162) | class MLPRender_Fea(torch.nn.Module):
    method __init__ (line 163) | def __init__(self, inChanel, num_layers=3, hidden_dim=64, viewpe=6, fe...
    method forward (line 188) | def forward(self, viewdirs, features):
  class FactorFields (line 206) | class FactorFields(torch.nn.Module):
    method __init__ (line 207) | def __init__(self, cfg, device):
    method setup_params (line 262) | def setup_params(self, aabb):
    method init_coef (line 311) | def init_coef(self):
    method init_basis (line 334) | def init_basis(self):
    method get_coeff (line 425) | def get_coeff(self, xyz_sampled):
    method get_basis (line 467) | def get_basis(self, x):
    method normalize_basis (line 519) | def normalize_basis(self):
    method get_coding (line 523) | def get_coding(self, x):
    method n_parameters (line 535) | def n_parameters(self):
    method get_optparam_groups (line 541) | def get_optparam_groups(self, lr_small=0.001, lr_large=0.02):
    method set_optimizable (line 556) | def set_optimizable(self, items, statue):
    method TV_loss (line 569) | def TV_loss(self, reg):
    method sample_point_ndc (line 575) | def sample_point_ndc(self, rays_o, rays_d, is_train=True, N_samples=-1):
    method sample_point (line 586) | def sample_point(self, rays_o, rays_d, is_train=True, N_samples=-1):
    method sample_point_unbound (line 604) | def sample_point_unbound(self, rays_o, rays_d, is_train=True, N_sample...
    method normalize_coord (line 635) | def normalize_coord(self, xyz_sampled):
    method basis2density (line 639) | def basis2density(self, density_features):
    method cal_mean_coef (line 646) | def cal_mean_coef(self, state_dict):
    method save (line 669) | def save(self, path):
    method load (line 682) | def load(self, ckpt):
    method update_renderParams (line 693) | def update_renderParams(self, gridSize):
    method upsample_volume_grid (line 702) | def upsample_volume_grid(self, res_target):
    method compute_alpha (line 710) | def compute_alpha(self, xyz_locs, length=1):
    method getDenseAlpha (line 730) | def getDenseAlpha(self, gridSize=None, times=16):
    method updateAlphaMask (line 758) | def updateAlphaMask(self, gridSize=(200, 200, 200), is_update_alphaMas...
    method shrink (line 796) | def shrink(self, new_aabb):
    method filtering_rays (line 812) | def filtering_rays(self, all_rays, all_rgbs, N_samples=256, chunk=1024...
    method forward (line 843) | def forward(self, rays_chunk, white_bg=True, is_train=False, ndc_ray=F...

FILE: models/sh.py
  function eval_sh (line 34) | def eval_sh(deg, sh, dirs):
  function eval_sh_bases (line 87) | def eval_sh_bases(deg, dirs):

FILE: renderer.py
  function render_ray (line 8) | def render_ray(rays, factor_fields, chunk=4096, N_samples=-1, ndc_ray=Fa...
  function evaluation (line 30) | def evaluation(test_dataset,factor_fields, renderer, savePath=None, N_vi...
  function evaluation_path (line 101) | def evaluation_path(test_dataset,factor_fields, c2ws, renderer, savePath...

FILE: run_batch.py
  function run_program (line 123) | def run_program(gpu, cmd):

FILE: scripts/2D_set_regression.py
  function PSNR (line 19) | def PSNR(a, b):
  function eval_img (line 29) | def eval_img(aabb, reso, idx, shiftment=[0.5, 0.5, 0.5], chunk=10240):
  function eval_img_single (line 49) | def eval_img_single(aabb, reso, chunk=10240):
  function linear_to_srgb (line 67) | def linear_to_srgb(img):
  function srgb_to_linear (line 72) | def srgb_to_linear(img):
  function write_image_imageio (line 77) | def write_image_imageio(img_file, img, colormap=None, quality=100):
  function interpolate (line 97) | def interpolate(colormap, x):

FILE: train_across_scene.py
  class SimpleSampler (line 17) | class SimpleSampler:
    method __init__ (line 18) | def __init__(self, total, batch):
    method nextids (line 24) | def nextids(self):
  function export_mesh (line 33) | def export_mesh(cfg):
  function render_test (line 55) | def render_test(cfg):
  function reconstruction (line 99) | def reconstruction(cfg):

FILE: train_across_scene_ft.py
  class SimpleSampler (line 17) | class SimpleSampler:
    method __init__ (line 18) | def __init__(self, total, batch):
    method nextids (line 24) | def nextids(self):
  function export_mesh (line 33) | def export_mesh(cfg):
  function render_test (line 42) | def render_test(cfg):
  function reconstruction (line 88) | def reconstruction(cfg):

FILE: train_per_scene.py
  class SimpleSampler (line 16) | class SimpleSampler:
    method __init__ (line 17) | def __init__(self, total, batch):
    method nextids (line 23) | def nextids(self):
  function export_mesh (line 32) | def export_mesh(ckpt_path):
  function render_test (line 44) | def render_test(cfg):
  function reconstruction (line 89) | def reconstruction(cfg):

FILE: utils.py
  function visualize_depth_numpy (line 11) | def visualize_depth_numpy(depth, minmax=None, cmap=cv2.COLORMAP_JET):
  function init_log (line 28) | def init_log(log, keys):
  function visualize_depth (line 33) | def visualize_depth(depth, minmax=None, cmap=cv2.COLORMAP_JET):
  function N_to_reso (line 53) | def N_to_reso(n_voxels, bbox):
  function N_to_vm_reso (line 59) | def N_to_vm_reso(n_voxels, bbox):
  function cal_n_samples (line 69) | def cal_n_samples(reso, step_ratio=0.5):
  class SimpleSampler (line 72) | class SimpleSampler:
    method __init__ (line 73) | def __init__(self, total, batch):
    method nextids (line 79) | def nextids(self):
  function init_lpips (line 87) | def init_lpips(net_name, device):
  function rgb_lpips (line 93) | def rgb_lpips(np_gt, np_im, net_name, device):
  function findItem (line 101) | def findItem(items, target):
  function rgb_ssim (line 110) | def rgb_ssim(img0, img1, max_val,
  class TVLoss (line 160) | class TVLoss(nn.Module):
    method __init__ (line 161) | def __init__(self,TVLoss_weight=1):
    method forward (line 165) | def forward(self,x):
    method _tensor_size (line 175) | def _tensor_size(self,t):
  function marchcude_to_world (line 178) | def marchcude_to_world(vertices, reso_WHD):
  function convert_sdf_samples_to_ply (line 183) | def convert_sdf_samples_to_ply(