SYMBOL INDEX (613 symbols across 81 files)

FILE: FourierGrid/FourierGrid_ckpt_manager.py
  class FourierGridCheckpointManager (line 11) | class FourierGridCheckpointManager:
    method __init__ (line 12) | def __init__(self, args, cfg) -> None:
    method load_all_info (line 17) | def load_all_info(self, model, optimizer, ckpt_path, no_reload_optimiz...
    method load_existing_model (line 25) | def load_existing_model(self, args, cfg, cfg_train, reload_ckpt_path, ...
    method save_model (line 44) | def save_model(self, global_step, model, optimizer, save_path):
    method load_model (line 53) | def load_model(self, model_class, ckpt_path):
    method merge_blocks (line 61) | def merge_blocks(self, args, cfg, device):

FILE: FourierGrid/FourierGrid_grid.py
  function create_grid (line 14) | def create_grid(type, **kwargs):
  class NeRFPosEmbedding (line 21) | class NeRFPosEmbedding(nn.Module):
    method __init__ (line 22) | def __init__(self, num_freqs: int, logscale=True):
    method forward (line 32) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class FourierGrid (line 42) | class FourierGrid(nn.Module):
    method __init__ (line 43) | def __init__(self, channels, world_size, xyz_min, xyz_max, use_nerf_po...
    method forward (line 60) | def forward(self, xyz):
    method scale_volume_grid (line 80) | def scale_volume_grid(self, new_world_size):
    method total_variation_add_grad (line 87) | def total_variation_add_grad(self, wx, wy, wz, dense_mode):
    method get_dense_grid (line 92) | def get_dense_grid(self):
    method __isub__ (line 96) | def __isub__(self, val):
    method extra_repr (line 100) | def extra_repr(self):
  function compute_tensorf_feat (line 104) | def compute_tensorf_feat(xy_plane, xz_plane, yz_plane, x_vec, y_vec, z_v...
  function compute_tensorf_val (line 122) | def compute_tensorf_val(xy_plane, xz_plane, yz_plane, x_vec, y_vec, z_ve...
  class MaskGrid (line 138) | class MaskGrid(nn.Module):
    method __init__ (line 139) | def __init__(self, path=None, mask_cache_thres=None, mask=None, xyz_mi...
    method forward (line 160) | def forward(self, xyz):
    method extra_repr (line 170) | def extra_repr(self):

FILE: FourierGrid/FourierGrid_model.py
  function get_rays (line 21) | def get_rays(H, W, K, c2w, inverse_y, flip_x, flip_y, mode='center'):
  function ndc_rays (line 52) | def ndc_rays(H, W, focal, near, rays_o, rays_d):
  function get_rays_of_a_view (line 72) | def get_rays_of_a_view(H, W, K, c2w, ndc, inverse_y, flip_x, flip_y, mod...
  class NeRFPosEmbedding (line 80) | class NeRFPosEmbedding(nn.Module):
    method __init__ (line 81) | def __init__(self, num_freqs: int, logscale=True):
    method forward (line 92) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class FourierMSELoss (line 100) | class FourierMSELoss(nn.Module):
    method __init__ (line 101) | def __init__(self,):
    method forward (line 104) | def forward(self, pred, gt):
    method __init__ (line 115) | def __init__(self, num_freqs=7, logscale=True):
    method forward (line 119) | def forward(self, pred, gt):
  class FourierMSELoss (line 114) | class FourierMSELoss(nn.Module):
    method __init__ (line 101) | def __init__(self,):
    method forward (line 104) | def forward(self, pred, gt):
    method __init__ (line 115) | def __init__(self, num_freqs=7, logscale=True):
    method forward (line 119) | def forward(self, pred, gt):
  class FourierGridModel (line 134) | class FourierGridModel(nn.Module):
    method __init__ (line 135) | def __init__(self, xyz_min, xyz_max, num_voxels_density=0, num_voxels_...
    method FourierGrid_get_training_rays (line 264) | def FourierGrid_get_training_rays(self, rgb_tr_ori, train_poses, HW, K...
    method gather_training_rays (line 297) | def gather_training_rays(self, data_dict, images, cfg, i_train, cfg_tr...
    method _set_grid_resolution (line 335) | def _set_grid_resolution(self, num_voxels_density, num_voxels_rgb):
    method get_kwargs (line 351) | def get_kwargs(self):
    method maskout_near_cam_vox (line 376) | def maskout_near_cam_vox(self, cam_o, near_clip):
    method voxel_count_views (line 390) | def voxel_count_views(self, rays_o_tr, rays_d_tr, imsz, near, far, ste...
    method scale_volume_grid (line 423) | def scale_volume_grid(self, num_voxels_density, num_voxels_rgb):
    method update_occupancy_cache (line 443) | def update_occupancy_cache(self):
    method update_occupancy_cache_lt_nviews (line 458) | def update_occupancy_cache_lt_nviews(self, rays_o_tr, rays_d_tr, imsz,...
    method density_total_variation_add_grad (line 482) | def density_total_variation_add_grad(self, weight, dense_mode):
    method k0_total_variation_add_grad (line 486) | def k0_total_variation_add_grad(self, weight, dense_mode):
    method activate_density (line 490) | def activate_density(self, density, interval=None):
    method hit_coarse_geo (line 495) | def hit_coarse_geo(self, rays_o, rays_d, near, far, stepsize, **render...
    method sample_ray (line 509) | def sample_ray(self, ori_rays_o, ori_rays_d, stepsize, is_train=False,...
    method forward (line 554) | def forward(self, rays_o, rays_d, viewdirs, global_step=None, is_train...
    method export_geometry_for_visualize (line 674) | def export_geometry_for_visualize(self, save_path):
  class DistortionLoss (line 684) | class DistortionLoss(torch.autograd.Function):
    method forward (line 686) | def forward(ctx, w, s, n_max, ray_id):
    method backward (line 698) | def backward(ctx, grad_back):

FILE: FourierGrid/arf.py
  class ARF (line 9) | class ARF:
    method __init__ (line 10) | def __init__(self, cfg, data_dict, device):
    method load_style_img (line 25) | def load_style_img(self, style_img_p):
    method match_colors_for_image_set (line 51) | def match_colors_for_image_set(self, image_set, train_save_dir):  # co...

FILE: FourierGrid/bbox_compute.py
  function _compute_bbox_by_cam_frustrm_unbounded (line 10) | def _compute_bbox_by_cam_frustrm_unbounded(cfg, HW, Ks, poses, i_train, ...
  function FourierGrid_compute_bbox_by_cam_frustrm_nerfpp (line 29) | def FourierGrid_compute_bbox_by_cam_frustrm_nerfpp(cfg, HW, Ks, poses, i...
  function FourierGrid_compute_bbox_by_cam_frustrm_waymo (line 48) | def FourierGrid_compute_bbox_by_cam_frustrm_waymo(cfg, HW, Ks, poses, i_...
  function FourierGrid_compute_bbox_by_cam_frustrm_mega (line 75) | def FourierGrid_compute_bbox_by_cam_frustrm_mega(cfg, HW, Ks, poses, i_t...
  function _compute_bbox_by_cam_frustrm_bounded (line 96) | def _compute_bbox_by_cam_frustrm_bounded(cfg, HW, Ks, poses, i_train, ne...
  function compute_bbox_by_cam_frustrm (line 113) | def compute_bbox_by_cam_frustrm(args, cfg, HW, Ks, poses, i_train, near,...
  function compute_bbox_by_coarse_geo (line 137) | def compute_bbox_by_coarse_geo(model_class, model_path, thres, device, a...

FILE: FourierGrid/camera_utils.py
  class Rays (line 10) | class Rays:
  function split_rays (line 19) | def split_rays(rays, batch_size):
  function intrinsic_matrix (line 32) | def intrinsic_matrix(fx: float,
  class ProjectionType (line 44) | class ProjectionType(enum.Enum):
  function convert_to_ndc (line 50) | def convert_to_ndc(origins: Tensor,
  function pixels_to_rays (line 118) | def pixels_to_rays(
  function _compute_residual_and_jacobian (line 232) | def _compute_residual_and_jacobian(
  function _radial_and_tangential_undistort (line 282) | def _radial_and_tangential_undistort(

FILE: FourierGrid/common_data_loaders/load_blendedmvs.py
  function load_blendedmvs_data (line 11) | def load_blendedmvs_data(basedir):

FILE: FourierGrid/common_data_loaders/load_blender.py
  function pose_spherical (line 30) | def pose_spherical(theta, phi, radius):
  function load_blender_data (line 38) | def load_blender_data(basedir, half_res=False, testskip=1):

FILE: FourierGrid/common_data_loaders/load_co3d.py
  function load_co3d_data (line 12) | def load_co3d_data(cfg):

FILE: FourierGrid/common_data_loaders/load_common_data.py
  function load_common_data (line 14) | def load_common_data(args):
  function inward_nearfar_heuristic (line 248) | def inward_nearfar_heuristic(cam_o, ratio=0.05):

FILE: FourierGrid/common_data_loaders/load_deepvoxels.py
  function load_dv_data (line 6) | def load_dv_data(scene='cube', basedir='/data/deepvoxels', testskip=1):

FILE: FourierGrid/common_data_loaders/load_free.py
  function imread (line 14) | def imread(f):
  function depthread (line 20) | def depthread(path):
  function _minify (line 38) | def _minify(basedir, factors=[], resolutions=[]):
  function normalize_scene (line 95) | def normalize_scene(poses, n_images, bounds):
  function load_images_from_disk (line 107) | def load_images_from_disk(basedir, factor, height, width):
  function normalize (line 146) | def normalize(x):
  function ptstocam (line 149) | def ptstocam(pts, c2w):
  function viewmatrix (line 153) | def viewmatrix(z, up, pos):
  function render_path_spiral (line 162) | def render_path_spiral(c2w, up, rads, focal, zdelta, zrate, rots, N):
  function poses_avg (line 174) | def poses_avg(poses):
  function recenter_poses (line 182) | def recenter_poses(poses, render_poses):
  function rerotate_poses (line 205) | def rerotate_poses(poses):
  function spherify_poses (line 233) | def spherify_poses(poses, bds, depths):
  function load_free_data (line 272) | def load_free_data(args, basedir, factor=8, width=None, height=None,

FILE: FourierGrid/common_data_loaders/load_llff.py
  function imread (line 12) | def imread(f):
  function depthread (line 18) | def depthread(path):
  function _minify (line 36) | def _minify(basedir, factors=[], resolutions=[]):
  function _load_data (line 93) | def _load_data(basedir, factor=None, width=None, height=None, load_imgs=...
  function normalize (line 182) | def normalize(x):
  function viewmatrix (line 185) | def viewmatrix(z, up, pos):
  function ptstocam (line 193) | def ptstocam(pts, c2w):
  function poses_avg (line 197) | def poses_avg(poses):
  function render_path_spiral (line 207) | def render_path_spiral(c2w, up, rads, focal, zdelta, zrate, rots, N):
  function recenter_poses (line 219) | def recenter_poses(poses):
  function rerotate_poses (line 233) | def rerotate_poses(poses):
  function spherify_poses (line 261) | def spherify_poses(poses, bds, depths):
  function load_llff_data (line 300) | def load_llff_data(basedir, factor=8, width=None, height=None,

FILE: FourierGrid/common_data_loaders/load_nerfpp.py
  function find_files (line 17) | def find_files(dir, exts):
  function load_data_split (line 29) | def load_data_split(split_dir, skip=1, try_load_min_depth=True, only_img...
  function rerotate_poses (line 89) | def rerotate_poses(poses, render_poses):
  function load_nerfpp_data (line 120) | def load_nerfpp_data(basedir, rerotate=True, training_ids=None):

FILE: FourierGrid/common_data_loaders/load_nsvf.py
  function pose_spherical (line 30) | def pose_spherical(theta, phi, radius):
  function load_nsvf_data (line 39) | def load_nsvf_data(basedir):

FILE: FourierGrid/common_data_loaders/load_tankstemple.py
  function normalize (line 11) | def normalize(x):
  function load_tankstemple_data (line 14) | def load_tankstemple_data(basedir, movie_render_kwargs={}):

FILE: FourierGrid/cuda/adam_upd.cpp
  function adam_upd (line 36) | void adam_upd(
  function masked_adam_upd (line 50) | void masked_adam_upd(
  function adam_upd_with_perlr (line 64) | void adam_upd_with_perlr(
  function PYBIND11_MODULE (line 79) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: FourierGrid/cuda/render_utils.cpp
  function infer_t_minmax (line 50) | std::vector<torch::Tensor> infer_t_minmax(
  function infer_n_samples (line 60) | torch::Tensor infer_n_samples(torch::Tensor rays_d, torch::Tensor t_min,...
  function infer_ray_start_dir (line 67) | std::vector<torch::Tensor> infer_ray_start_dir(torch::Tensor rays_o, tor...
  function sample_pts_on_rays (line 74) | std::vector<torch::Tensor> sample_pts_on_rays(
  function sample_ndc_pts_on_rays (line 87) | std::vector<torch::Tensor> sample_ndc_pts_on_rays(
  function sample_bg_pts_on_rays (line 100) | torch::Tensor sample_bg_pts_on_rays(
  function maskcache_lookup (line 109) | torch::Tensor maskcache_lookup(torch::Tensor world, torch::Tensor xyz, t...
  function raw2alpha (line 120) | std::vector<torch::Tensor> raw2alpha(torch::Tensor density, const float ...
  function raw2alpha_nonuni (line 125) | std::vector<torch::Tensor> raw2alpha_nonuni(torch::Tensor density, const...
  function raw2alpha_backward (line 131) | torch::Tensor raw2alpha_backward(torch::Tensor exp, torch::Tensor grad_b...
  function raw2alpha_nonuni_backward (line 136) | torch::Tensor raw2alpha_nonuni_backward(torch::Tensor exp, torch::Tensor...
  function alpha2weight (line 142) | std::vector<torch::Tensor> alpha2weight(torch::Tensor alpha, torch::Tens...
  function alpha2weight_backward (line 151) | torch::Tensor alpha2weight_backward(
  function PYBIND11_MODULE (line 170) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: FourierGrid/cuda/setup.py
  function setup_sources (line 7) | def setup_sources(name, sources):

FILE: FourierGrid/cuda/total_variation.cpp
  function total_variation_add_grad (line 16) | void total_variation_add_grad(torch::Tensor param, torch::Tensor grad, f...
  function PYBIND11_MODULE (line 22) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: FourierGrid/cuda/ub360_utils.cpp
  function cumdist_thres (line 15) | torch::Tensor cumdist_thres(torch::Tensor dist, float thres) {
  function PYBIND11_MODULE (line 20) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: FourierGrid/dcvgo.py
  class DirectContractedVoxGO (line 28) | class DirectContractedVoxGO(nn.Module):
    method __init__ (line 29) | def __init__(self, xyz_min, xyz_max,
    method _set_grid_resolution (line 127) | def _set_grid_resolution(self, num_voxels):
    method get_kwargs (line 139) | def get_kwargs(self):
    method scale_volume_grid (line 158) | def scale_volume_grid(self, num_voxels):
    method update_occupancy_cache (line 181) | def update_occupancy_cache(self):
    method update_occupancy_cache_lt_nviews (line 195) | def update_occupancy_cache_lt_nviews(self, rays_o_tr, rays_d_tr, imsz,...
    method density_total_variation_add_grad (line 215) | def density_total_variation_add_grad(self, weight, dense_mode):
    method k0_total_variation_add_grad (line 219) | def k0_total_variation_add_grad(self, weight, dense_mode):
    method activate_density (line 223) | def activate_density(self, density, interval=None):
    method sample_ray (line 228) | def sample_ray(self, ori_rays_o, ori_rays_d, stepsize, is_train=False,...
    method forward (line 264) | def forward(self, rays_o, rays_d, viewdirs, global_step=None, is_train...
  class DistortionLoss (line 387) | class DistortionLoss(torch.autograd.Function):
    method forward (line 389) | def forward(ctx, w, s, n_max, ray_id):
    method backward (line 401) | def backward(ctx, grad_back):

FILE: FourierGrid/dmpigo.py
  class DirectMPIGO (line 18) | class DirectMPIGO(torch.nn.Module):
    method __init__ (line 19) | def __init__(self, xyz_min, xyz_max,
    method _set_grid_resolution (line 120) | def _set_grid_resolution(self, num_voxels, mpi_depth):
    method get_kwargs (line 132) | def get_kwargs(self):
    method scale_volume_grid (line 151) | def scale_volume_grid(self, num_voxels, mpi_depth):
    method update_occupancy_cache (line 175) | def update_occupancy_cache(self):
    method update_occupancy_cache_lt_nviews (line 189) | def update_occupancy_cache_lt_nviews(self, rays_o_tr, rays_d_tr, imsz,...
    method density_total_variation_add_grad (line 209) | def density_total_variation_add_grad(self, weight, dense_mode):
    method k0_total_variation_add_grad (line 214) | def k0_total_variation_add_grad(self, weight, dense_mode):
    method activate_density (line 219) | def activate_density(self, density, interval=None):
    method sample_ray (line 224) | def sample_ray(self, rays_o, rays_d, near, far, stepsize, **render_kwa...
    method forward (line 251) | def forward(self, rays_o, rays_d, viewdirs, global_step=None, **render...
  function create_full_step_id (line 344) | def create_full_step_id(shape):

FILE: FourierGrid/dvgo.py
  class DirectVoxGO (line 26) | class DirectVoxGO(torch.nn.Module):
    method __init__ (line 27) | def __init__(self, xyz_min, xyz_max,
    method _set_grid_resolution (line 154) | def _set_grid_resolution(self, num_voxels):
    method get_kwargs (line 165) | def get_kwargs(self):
    method maskout_near_cam_vox (line 185) | def maskout_near_cam_vox(self, cam_o, near_clip):
    method scale_volume_grid (line 213) | def scale_volume_grid(self, num_voxels):
    method update_occupancy_cache (line 236) | def update_occupancy_cache(self):
    method voxel_count_views (line 247) | def voxel_count_views(self, rays_o_tr, rays_d_tr, imsz, near, far, ste...
    method density_total_variation_add_grad (line 279) | def density_total_variation_add_grad(self, weight, dense_mode):
    method k0_total_variation_add_grad (line 283) | def k0_total_variation_add_grad(self, weight, dense_mode):
    method activate_density (line 287) | def activate_density(self, density, interval=None):
    method hit_coarse_geo (line 292) | def hit_coarse_geo(self, rays_o, rays_d, near, far, stepsize, **render...
    method sample_ray (line 306) | def sample_ray(self, rays_o, rays_d, near, far, stepsize, **render_kwa...
    method forward (line 330) | def forward(self, rays_o, rays_d, viewdirs, global_step=None, **render...
  class Raw2Alpha (line 430) | class Raw2Alpha(torch.autograd.Function):
    method forward (line 432) | def forward(ctx, density, shift, interval):
    method backward (line 447) | def backward(ctx, grad_back):
  class Raw2Alpha_nonuni (line 456) | class Raw2Alpha_nonuni(torch.autograd.Function):
    method forward (line 458) | def forward(ctx, density, shift, interval):
    method backward (line 467) | def backward(ctx, grad_back):
  class Alphas2Weights (line 472) | class Alphas2Weights(torch.autograd.Function):
    method forward (line 474) | def forward(ctx, alpha, ray_id, N):
    method backward (line 483) | def backward(ctx, grad_weights, grad_last):
  function get_rays (line 493) | def get_rays(H, W, K, c2w, inverse_y, flip_x, flip_y, mode='center'):
  function get_rays_np (line 524) | def get_rays_np(H, W, K, c2w):
  function ndc_rays (line 534) | def ndc_rays(H, W, focal, near, rays_o, rays_d):
  function get_rays_of_a_view (line 554) | def get_rays_of_a_view(H, W, K, c2w, ndc, inverse_y, flip_x, flip_y, mod...
  function get_training_rays (line 563) | def get_training_rays(rgb_tr, train_poses, HW, Ks, ndc, inverse_y, flip_...
  function get_training_rays_flatten (line 588) | def get_training_rays_flatten(rgb_tr_ori, train_poses, HW, Ks, ndc, inve...
  function get_training_rays_in_maskcache_sampling (line 620) | def get_training_rays_in_maskcache_sampling(rgb_tr_ori, train_poses, HW,...
  function batch_indices_generator (line 660) | def batch_indices_generator(N, BS):

FILE: FourierGrid/grid.py
  function create_grid (line 30) | def create_grid(type, **kwargs):
  class DenseGrid (line 41) | class DenseGrid(nn.Module):
    method __init__ (line 42) | def __init__(self, channels, world_size, xyz_min, xyz_max, **kwargs):
    method forward (line 50) | def forward(self, xyz):
    method scale_volume_grid (line 63) | def scale_volume_grid(self, new_world_size):
    method total_variation_add_grad (line 70) | def total_variation_add_grad(self, wx, wy, wz, dense_mode):
    method get_dense_grid (line 75) | def get_dense_grid(self):
    method __isub__ (line 79) | def __isub__(self, val):
    method extra_repr (line 83) | def extra_repr(self):
  class TensoRFGrid (line 90) | class TensoRFGrid(nn.Module):
    method __init__ (line 91) | def __init__(self, channels, world_size, xyz_min, xyz_max, config):
    method forward (line 111) | def forward(self, xyz):
    method scale_volume_grid (line 131) | def scale_volume_grid(self, new_world_size):
    method total_variation_add_grad (line 142) | def total_variation_add_grad(self, wx, wy, wz, dense_mode):
    method get_dense_grid (line 156) | def get_dense_grid(self):
    method extra_repr (line 171) | def extra_repr(self):
  function compute_tensorf_feat (line 174) | def compute_tensorf_feat(xy_plane, xz_plane, yz_plane, x_vec, y_vec, z_v...
  function compute_tensorf_val (line 191) | def compute_tensorf_val(xy_plane, xz_plane, yz_plane, x_vec, y_vec, z_ve...
  class MaskGrid (line 207) | class MaskGrid(nn.Module):
    method __init__ (line 208) | def __init__(self, path=None, mask_cache_thres=None, mask=None, xyz_mi...
    method forward (line 231) | def forward(self, xyz):
    method extra_repr (line 241) | def extra_repr(self):

FILE: FourierGrid/load_everything.py
  function load_everything (line 9) | def load_everything(args, cfg):
  function load_existing_model (line 41) | def load_existing_model(args, cfg, cfg_train, reload_ckpt_path, device):

FILE: FourierGrid/load_mega.py
  function find_files (line 24) | def find_files(dir, exts):
  function waymo_load_img_list (line 36) | def waymo_load_img_list(split_dir, skip=1):
  function sample_list_by_idx (line 47) | def sample_list_by_idx(one_list, idxs):
  function sample_metadata_by_cam (line 52) | def sample_metadata_by_cam(metadata, cam_idx):
  function find_most_freq_ele (line 63) | def find_most_freq_ele(one_list):
  function sample_metadata_by_shape (line 69) | def sample_metadata_by_shape(metadata):
  function sample_metadata_by_idxs (line 82) | def sample_metadata_by_idxs(metadata, sample_idxs):
  function sort_metadata_by_pos (line 91) | def sort_metadata_by_pos(metadata):
  function normalize (line 109) | def normalize(x):
  function viewmatrix (line 113) | def viewmatrix(z, up, pos):
  function poses_avg (line 122) | def poses_avg(poses):
  function sample_metadata_by_training_ids (line 131) | def sample_metadata_by_training_ids(metadata, training_ids, assign_pos, ...
  function load_mega (line 163) | def load_mega(args, cfg, ):
  function inward_nearfar_heuristic (line 248) | def inward_nearfar_heuristic(cam_o, ratio=0.05):
  function load_mega_data (line 257) | def load_mega_data(args, cfg):

FILE: FourierGrid/load_waymo.py
  function find_files (line 25) | def find_files(dir, exts):
  function waymo_load_img_list (line 37) | def waymo_load_img_list(split_dir, skip=1):
  function rerotate_poses (line 47) | def rerotate_poses(poses, render_poses):
  function sample_list_by_idx (line 78) | def sample_list_by_idx(one_list, idxs):
  function sample_metadata_by_cam (line 83) | def sample_metadata_by_cam(metadata, cam_idx):
  function sample_metadata_by_idxs (line 94) | def sample_metadata_by_idxs(metadata, sample_idxs, val_num=5):
  function sample_metadata_by_training_ids (line 110) | def sample_metadata_by_training_ids(metadata, training_ids, assign_pos, ...
  function sort_metadata_by_pos (line 142) | def sort_metadata_by_pos(metadata):
  function normalize (line 153) | def normalize(x):
  function viewmatrix (line 157) | def viewmatrix(z, up, pos):
  function ptstocam (line 166) | def ptstocam(pts, c2w):
  function poses_avg (line 171) | def poses_avg(poses):
  function recenter_poses (line 180) | def recenter_poses(poses):
  function find_most_freq_ele (line 194) | def find_most_freq_ele(one_list):
  function save_training_imgs_to_disk (line 200) | def save_training_imgs_to_disk(args, cfg, metadata):
  function resize_img (line 213) | def resize_img(train_HW, val_HW, imgs, tr_K, val_K):
  function find_rotations_from_meta (line 254) | def find_rotations_from_meta(metadata):
  function load_waymo (line 263) | def load_waymo(args, cfg, ):
  function inward_nearfar_heuristic (line 371) | def inward_nearfar_heuristic(cam_o, ratio=0.05):
  function load_waymo_data (line 380) | def load_waymo_data(args, cfg):

FILE: FourierGrid/masked_adam.py
  class MaskedAdam (line 21) | class MaskedAdam(torch.optim.Optimizer):
    method __init__ (line 23) | def __init__(self, params, lr=1e-3, betas=(0.9, 0.99), eps=1e-8):
    method __setstate__ (line 36) | def __setstate__(self, state):
    method set_pervoxel_lr (line 39) | def set_pervoxel_lr(self, count):
    method step (line 44) | def step(self):

FILE: FourierGrid/pose_utils/image_operators.py
  function get_bbox_from_img (line 7) | def get_bbox_from_img(image, color_thre=1e-2):
  function change_background_from_black_to_white (line 19) | def change_background_from_black_to_white(image, color_thresh=1e-2):
  function get_bbox_from_mask (line 25) | def get_bbox_from_mask(label_img):
  function apply_mask_on_img (line 33) | def apply_mask_on_img(one_img, label_img):
  function image_normalization_for_pose (line 41) | def image_normalization_for_pose(image):

FILE: FourierGrid/pose_utils/linemod_evaluator.py
  function rotation_angle_euler (line 9) | def rotation_angle_euler(R1, R2):
  function project (line 26) | def project(xyz, K, RT):
  class LineMODEvaluator (line 38) | class LineMODEvaluator:
    method __init__ (line 39) | def __init__(self, class_name, obj_m, icp_refine=False):
    method projection_2d (line 71) | def projection_2d(self, pose_pred, pose_targets, K, icp=False, thresho...
    method projection_2d_sym (line 81) | def projection_2d_sym(self, pose_pred, pose_targets, K, threshold=5):
    method add2_metric (line 89) | def add2_metric(self, pose_pred, pose_targets, icp=False, syn=False, p...
    method add5_metric (line 110) | def add5_metric(self, pose_pred, pose_targets, icp=False, syn=False, p...
    method add_metric (line 130) | def add_metric(self, pose_pred, pose_targets, icp=False, syn=False, pe...
    method cm_degree_5_metric (line 162) | def cm_degree_5_metric(self, pose_pred, pose_targets, icp=False):
    method mask_iou (line 175) | def mask_iou(self, output, batch):
    method icp_refine (line 182) | def icp_refine(self, pose_pred, anno, output, K):
    method icp_refine_ (line 200) | def icp_refine_(self, pose, anno, output):
    method summarize (line 241) | def summarize(self):
    method evaluate_proposals (line 283) | def evaluate_proposals(self, pose_gt, pose_proposals, cam_k):
    method evaluate_linemod (line 303) | def evaluate_linemod(self, pose_gt, pose_pred, cam_k): # sample_corres...

FILE: FourierGrid/pose_utils/model_operations.py
  function get_bb8_of_model (line 5) | def get_bb8_of_model(obj_m):

FILE: FourierGrid/pose_utils/pose_operators.py
  function chordal_distance (line 6) | def chordal_distance(R1,R2):
  function rotation_angle_chordal (line 10) | def rotation_angle_chordal(R1, R2):
  function cal_pose_rot_diff (line 14) | def cal_pose_rot_diff(pose1, pose2):
  function rot_diff_to_norm_angle (line 19) | def rot_diff_to_norm_angle(rotation_difference):
  function cal_one_add (line 27) | def cal_one_add(model_points, pose_pred, pose_targets, syn=False):
  function pose_rot_interpolation (line 42) | def pose_rot_interpolation(pose_a, pose_b, inter_num=100):
  function pose_to_blender (line 67) | def pose_to_blender(pose):

FILE: FourierGrid/pose_utils/projection.py
  function get_projected_points (line 6) | def get_projected_points(cam_pose, cam_k, obj_m, one_img=None, save_root...

FILE: FourierGrid/pose_utils/visualization.py
  function visualize_2d_points (line 9) | def visualize_2d_points(points_2d, bg_image, save_root=None, pre_str="",...
  function get_projected_points (line 29) | def get_projected_points(cam_pose, cam_k, obj_m, one_img=None, save_root...
  function draw_bbox_8_2D (line 34) | def draw_bbox_8_2D(draw_img, bbox_8_2D, color = (0, 255, 0), thickness =...
  function visualize_pose_prediction (line 67) | def visualize_pose_prediction(pose_a, pose_b, cam_k, obj_bb8, bg_img, sa...

FILE: FourierGrid/pycolmap/pycolmap/camera.py
  function simple_radial_distortion (line 14) | def simple_radial_distortion(camera, x):
  function radial_distortion (line 17) | def radial_distortion(camera, x):
  function opencv_distortion (line 21) | def opencv_distortion(camera, x):
  class Camera (line 38) | class Camera:
    method GetNumParams (line 40) | def GetNumParams(type_):
    method GetNameFromType (line 71) | def GetNameFromType(type_):
    method __init__ (line 89) | def __init__(self, type_, width_, height_, params):
    method __str__ (line 128) | def __str__(self):
    method get_params (line 152) | def get_params(self):
    method get_camera_matrix (line 168) | def get_camera_matrix(self):
    method get_inverse_camera_matrix (line 172) | def get_inverse_camera_matrix(self):
    method K (line 179) | def K(self):
    method K_inv (line 183) | def K_inv(self):
    method get_inv_camera_matrix (line 189) | def get_inv_camera_matrix(self):
    method get_image_grid (line 199) | def get_image_grid(self):
    method distort_points (line 213) | def distort_points(self, x, normalized=True, denormalize=True):
    method undistort_points (line 237) | def undistort_points(self, x, normalized=False, denormalize=True):

FILE: FourierGrid/pycolmap/pycolmap/database.py
  function array_to_blob (line 9) | def array_to_blob(arr):
  function blob_to_array (line 12) | def blob_to_array(blob, dtype, shape=(-1,)):
  function get_pair_id (line 21) | def get_pair_id(image_id1, image_id2):
  function get_image_ids_from_pair_id (line 27) | def get_image_ids_from_pair_id(pair_id):
  function add_camera (line 98) | def add_camera(db, model, width, height, params, prior_focal_length=False,
  function add_descriptors (line 107) | def add_descriptors(db, image_id, descriptors):
  function add_image (line 113) | def add_image(db, name, camera_id, prior_q=np.zeros(4), prior_t=np.zeros...
  function add_inlier_matches (line 121) | def add_inlier_matches(db, image_id1, image_id2, matches, config=2, F=None,
  function add_keypoints (line 142) | def add_keypoints(db, image_id, keypoints):
  function add_matches (line 152) | def add_matches(db, image_id1, image_id2, matches):
  class COLMAPDatabase (line 168) | class COLMAPDatabase(sqlite3.Connection):
    method connect (line 170) | def connect(database_path):
    method __init__ (line 174) | def __init__(self, *args, **kwargs):
  function main (line 205) | def main(args):

FILE: FourierGrid/pycolmap/pycolmap/image.py
  class Image (line 11) | class Image:
    method __init__ (line 12) | def __init__(self, name_, camera_id_, q_, tvec_):
    method R (line 23) | def R(self):
    method C (line 28) | def C(self):
    method t (line 34) | def t(self):

FILE: FourierGrid/pycolmap/pycolmap/rotation.py
  function cross_prod_matrix (line 12) | def cross_prod_matrix(v):
  function axis_angle_to_rotation_matrix (line 20) | def axis_angle_to_rotation_matrix(axis, angle=None):
  function rotation_matrix_to_axis_angle (line 34) | def rotation_matrix_to_axis_angle(R):
  class Quaternion (line 43) | class Quaternion:
    method FromR (line 47) | def FromR(R):
    method FromAxisAngle (line 79) | def FromAxisAngle(axis, angle=None):
    method __init__ (line 92) | def __init__(self, q=np.array((1., 0., 0., 0.))):
    method __add__ (line 105) | def __add__(self, other):
    method __iadd__ (line 108) | def __iadd__(self, other):
    method __invert__ (line 113) | def __invert__(self):
    method __mul__ (line 119) | def __mul__(self, other):
    method __rmul__ (line 133) | def __rmul__(self, other):
    method __imul__ (line 136) | def __imul__(self, other):
    method __irmul__ (line 140) | def __irmul__(self, other):
    method __neg__ (line 144) | def __neg__(self):
    method __sub__ (line 147) | def __sub__(self, other):
    method __isub__ (line 150) | def __isub__(self, other):
    method __str__ (line 154) | def __str__(self):
    method copy (line 157) | def copy(self):
    method dot (line 160) | def dot(self, other):
    method inverse (line 164) | def inverse(self):
    method norm (line 167) | def norm(self):
    method normalize (line 170) | def normalize(self):
    method rotate_points (line 175) | def rotate_points(self, x):
    method ToR (line 180) | def ToR(self):
    method ToAxisAngle (line 193) | def ToAxisAngle(self):
    method ToEulerAngles (line 220) | def ToEulerAngles(self):
  class DualQuaternion (line 243) | class DualQuaternion:
    method FromQT (line 246) | def FromQT(q, t):
    method __init__ (line 249) | def __init__(self, q0=np.array((1., 0., 0., 0.)), qe=np.zeros(4)):
    method __add__ (line 252) | def __add__(self, other):
    method __iadd__ (line 255) | def __iadd__(self, other):
    method __invert__ (line 261) | def __invert__(self):
    method __mul__ (line 264) | def __mul__(self, other):
    method __rmul__ (line 276) | def __rmul__(self, other):
    method __imul__ (line 279) | def __imul__(self, other):
    method __neg__ (line 284) | def __neg__(self):
    method __sub__ (line 287) | def __sub__(self, other):
    method __isub__ (line 290) | def __isub__(self, other):
    method inverse (line 297) | def inverse(self):
    method norm (line 305) | def norm(self):
    method normalize (line 310) | def normalize(self):
    method getT (line 320) | def getT(self):
    method ToQT (line 323) | def ToQT(self):

FILE: FourierGrid/pycolmap/pycolmap/scene_manager.py
  class SceneManager (line 21) | class SceneManager:
    method __init__ (line 24) | def __init__(self, colmap_results_folder, image_path=None):
    method load_colmap_project_file (line 56) | def load_colmap_project_file(self, project_file=None, image_path=None):
    method load (line 79) | def load(self):
    method load_cameras (line 86) | def load_cameras(self, input_file=None):
    method _load_cameras_bin (line 98) | def _load_cameras_bin(self, input_file):
    method _load_cameras_txt (line 111) | def _load_cameras_txt(self, input_file):
    method load_images (line 127) | def load_images(self, input_file=None):
    method _load_images_bin (line 139) | def _load_images_bin(self, input_file):
    method _load_images_txt (line 181) | def _load_images_txt(self, input_file):
    method load_points3D (line 217) | def load_points3D(self, input_file=None):
    method _load_points3D_bin (line 229) | def _load_points3D_bin(self, input_file):
    method _load_points3D_txt (line 257) | def _load_points3D_txt(self, input_file):
    method save (line 290) | def save(self, output_folder, binary=True):
    method save_cameras (line 297) | def save_cameras(self, output_folder, output_file=None, binary=True):
    method _save_cameras_bin (line 311) | def _save_cameras_bin(self, output_file):
    method _save_cameras_txt (line 323) | def _save_cameras_txt(self, output_file):
    method save_images (line 334) | def save_images(self, output_folder, output_file=None, binary=True):
    method _save_images_bin (line 348) | def _save_images_bin(self, output_file):
    method _save_images_txt (line 363) | def _save_images_txt(self, output_file):
    method save_points3D (line 387) | def save_points3D(self, output_folder, output_file=None, binary=True):
    method _save_points3D_bin (line 401) | def _save_points3D_bin(self, output_file):
    method _save_points3D_txt (line 424) | def _save_points3D_txt(self, output_file):
    method get_image_from_name (line 455) | def get_image_from_name(self, image_name):
    method get_camera (line 461) | def get_camera(self, camera_id):
    method get_points3D (line 466) | def get_points3D(self, image_id, return_points2D=True, return_colors=F...
    method point3D_valid (line 489) | def point3D_valid(self, point3D_id):
    method get_filtered_points3D (line 495) | def get_filtered_points3D(self, return_colors=False):
    method get_shared_points3D (line 509) | def get_shared_points3D(self, image_id1, image_id2):
    method get_viewed_points (line 524) | def get_viewed_points(self, image_id):
    method add_camera (line 554) | def add_camera(self, camera):
    method add_image (line 561) | def add_image(self, image):
    method delete_images (line 568) | def delete_images(self, image_list):
    method filter_points3D (line 599) | def filter_points3D(self,
    method build_scene_graph (line 660) | def build_scene_graph(self):

FILE: FourierGrid/pycolmap/tools/colmap_to_nvm.py
  function main (line 12) | def main(args):

FILE: FourierGrid/pycolmap/tools/delete_images.py
  function main (line 11) | def main(args):

FILE: FourierGrid/pycolmap/tools/impute_missing_cameras.py
  function interpolate_linear (line 16) | def interpolate_linear(images, camera_id, file_format):
  function interpolate_hermite (line 54) | def interpolate_hermite(images, camera_id, file_format):
  function main (line 137) | def main(args):

FILE: FourierGrid/pycolmap/tools/save_cameras_as_ply.py
  function save_camera_ply (line 19) | def save_camera_ply(ply_file, images, scale):
  function main (line 64) | def main(args):

FILE: FourierGrid/pycolmap/tools/transform_model.py
  function main (line 11) | def main(args):

FILE: FourierGrid/pycolmap/tools/write_camera_track_to_bundler.py
  function main (line 11) | def main(args):

FILE: FourierGrid/pycolmap/tools/write_depthmap_to_ply.py
  function main (line 15) | def main(args):

FILE: FourierGrid/run_colmap2standard.py
  class NeRFSceneManager (line 14) | class NeRFSceneManager(SceneManager):
    method __init__ (line 21) | def __init__(self, data_dir):
    method process (line 24) | def process(
  class Dataset (line 115) | class Dataset:
    method __init__ (line 116) | def __init__(self, data_dir):
    method export (line 179) | def export(self, data_dir, out_mode):
  function main (line 218) | def main(data_dir, out_mode):

FILE: FourierGrid/run_export_bbox.py
  function run_export_bbox_cams (line 9) | def run_export_bbox_cams(args, cfg, data_dict, save_path=None):

FILE: FourierGrid/run_export_coarse.py
  function run_export_coarse (line 7) | def run_export_coarse(args, cfg, device, save_path=None):

FILE: FourierGrid/run_gen_cam_paths.py
  function select_k_nearest_points (line 15) | def select_k_nearest_points(idx, positions, k):
  function move_idxs_to_folder (line 23) | def move_idxs_to_folder(data_dict, sampled_idxs, save_path, data_root="d...
  function render_idxs (line 33) | def render_idxs(data_dict, sampled_idxs, save_path, data_root="data/sep1...
  function get_rotation_kp_2d (line 44) | def get_rotation_kp_2d(args, cfg, data_dict, sample_idxs):
  function run_export_bbox_cams (line 63) | def run_export_bbox_cams(args, cfg, data_dict, sample_idxs, save_path):
  function run_gen_cam_paths (line 87) | def run_gen_cam_paths(args, cfg, data_dict, core_cam=None, straight_leng...

FILE: FourierGrid/run_gtk_analysis.py
  class VoxelGrid (line 18) | class VoxelGrid(nn.Module):
    method __init__ (line 20) | def __init__(self, grid_len=1000, data_point_num=100):
    method forward (line 39) | def forward(self,):  # calculate GTK
    method one_d_regress (line 54) | def one_d_regress(self, x_train, x_test, y_train, y_test_gt):
  class FourierGrid (line 84) | class FourierGrid(nn.Module):
    method __init__ (line 86) | def __init__(self, grid_len=1000, band_num=10, data_point_num=100):
    method gamma_x_i (line 106) | def gamma_x_i(self, x, i):
    method forward (line 114) | def forward(self,):  # calculate GTK
    method one_d_regress (line 130) | def one_d_regress(self, x_train, x_test, y_train, y_test_gt):
  function train_model (line 170) | def train_model(one_model):
  function get_fg_gtk_spectrum_by_band_num (line 184) | def get_fg_gtk_spectrum_by_band_num(band_num):
  function sample_random_signal (line 263) | def sample_random_signal(key, decay_vec):
  function sample_random_powerlaw (line 272) | def sample_random_powerlaw(key, N, power):
  function get_sine_signal (line 280) | def get_sine_signal():
  function get_bessel_signal (line 284) | def get_bessel_signal():
  function calculate_Delta (line 333) | def calculate_Delta(gtk, y1_data, y2_data):

FILE: FourierGrid/run_render.py
  function render_viewpoints (line 15) | def render_viewpoints(cfg, model, render_poses, HW, Ks, ndc, render_kwargs,
  function run_render (line 117) | def run_render(args, cfg, data_dict, device, debug=True, add_info="", vi...

FILE: FourierGrid/run_train.py
  function create_new_model (line 19) | def create_new_model(args, cfg, cfg_model, cfg_train, xyz_min, xyz_max, ...
  function gather_training_rays (line 59) | def gather_training_rays(data_dict, images, cfg, i_train, cfg_train, pos...
  function scene_rep_reconstruction (line 98) | def scene_rep_reconstruction(args, cfg, cfg_model, cfg_train, xyz_min, x...
  function run_train (line 336) | def run_train(args, cfg, data_dict, export_cam=False, export_geometry=Fa...

FILE: FourierGrid/tools/colmap_utils/colmap_read_model.py
  class Image (line 49) | class Image(BaseImage):
    method qvec2rotmat (line 50) | def qvec2rotmat(self):
  function read_next_bytes (line 71) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact...
  function read_cameras_text (line 83) | def read_cameras_text(path):
  function read_cameras_binary (line 109) | def read_cameras_binary(path_to_model_file):
  function read_images_text (line 138) | def read_images_text(path):
  function read_images_binary (line 169) | def read_images_binary(path_to_model_file):
  function read_points3D_text (line 204) | def read_points3D_text(path):
  function read_points3d_binary (line 231) | def read_points3d_binary(path_to_model_file):
  function read_model (line 261) | def read_model(path, ext):
  function qvec2rotmat (line 273) | def qvec2rotmat(qvec):
  function rotmat2qvec (line 286) | def rotmat2qvec(R):
  function main (line 300) | def main():

FILE: FourierGrid/tools/colmap_utils/colmap_wrapper.py
  function run_colmap (line 24) | def run_colmap(basedir, match_type):

FILE: FourierGrid/tools/colmap_utils/pose_utils.py
  function load_colmap_data_nerfstudio (line 14) | def load_colmap_data_nerfstudio(basedir):
  function load_colmap_data (line 64) | def load_colmap_data(realdir):
  function save_poses (line 110) | def save_poses(basedir, poses, pts3d, perm, names):
  function minify (line 146) | def minify(basedir, factors=[], resolutions=[]):
  function gen_poses (line 199) | def gen_poses(basedir, match_type, factors=None):

FILE: FourierGrid/tools/imgs2poses.py
  function check_structure (line 9) | def check_structure(scenedir):

FILE: FourierGrid/tools/vis_volume.py
  function change_background_to_black (line 67) | def change_background_to_black(vis):

FILE: FourierGrid/trajectory_generators/interp_traj.py
  function inter_two_poses (line 11) | def inter_two_poses(pose_a, pose_b, alpha):
  function inter_poses (line 24) | def inter_poses(key_poses, n_out_poses, sigma=1.):
  function hello (line 46) | def hello(data_dir, n_out_poses, key_poses):

FILE: FourierGrid/trajectory_generators/mega_traj.py
  function all_equal (line 7) | def all_equal(iterable):
  function rotate_rot_matrix_by_degree (line 12) | def rotate_rot_matrix_by_degree(rot, rot_degree, axis='y'):
  function gen_dummy_trajs (line 18) | def gen_dummy_trajs(metadata, tr_c2w, train_HW, tr_K, test_num=100):
  function gen_straight_trajs (line 28) | def gen_straight_trajs(metadata, tr_c2w, train_HW, tr_K, tr_cam_idx, tra...
  function gen_rotational_trajs (line 44) | def gen_rotational_trajs(args, cfg, metadata, tr_c2w, train_HW, tr_K, ro...

FILE: FourierGrid/trajectory_generators/waymo_traj.py
  function all_equal (line 7) | def all_equal(iterable):
  function rotate_rot_matrix_by_degree (line 12) | def rotate_rot_matrix_by_degree(rot, rot_degree, axis='y'):
  function gen_dummy_trajs (line 18) | def gen_dummy_trajs(metadata, tr_c2w, train_HW, tr_K, test_num=100):
  function gen_straight_trajs (line 28) | def gen_straight_trajs(metadata, tr_c2w, train_HW, tr_K, tr_cam_idx, tra...
  function gen_rotational_trajs (line 44) | def gen_rotational_trajs(args, cfg, metadata, tr_c2w, train_HW, tr_K, tr...

FILE: FourierGrid/utils.py
  function resize_and_to_8b (line 21) | def resize_and_to_8b(input_images, res):
  function create_optimizer_or_freeze_model (line 26) | def create_optimizer_or_freeze_model(model, cfg_train, global_step, verb...
  function load_checkpoint (line 61) | def load_checkpoint(model, optimizer, ckpt_path, no_reload_optimizer):
  function load_model (line 70) | def load_model(model_class, ckpt_path):
  function rgb_ssim (line 79) | def rgb_ssim(img0, img1, max_val,
  function init_lpips (line 129) | def init_lpips(net_name, device):
  function rgb_lpips (line 135) | def rgb_lpips(np_gt, np_im, net_name, device):

FILE: block_nerf/block_nerf_lightning.py
  class Block_NeRF_System (line 13) | class Block_NeRF_System(LightningModule):
    method __init__ (line 14) | def __init__(self, hparams):
    method forward (line 45) | def forward(self, rays, ts):
    method setup (line 71) | def setup(self, stage):
    method configure_optimizers (line 85) | def configure_optimizers(self):
    method train_dataloader (line 90) | def train_dataloader(self):
    method val_dataloader (line 97) | def val_dataloader(self):
    method training_step (line 104) | def training_step(self, batch, batch_nb):
    method validation_step (line 121) | def validation_step(self, batch, batch_nb):  # validate at each epoch
    method validation_epoch_end (line 147) | def validation_epoch_end(self, outputs):

FILE: block_nerf/block_nerf_model.py
  class BlockNeRFLoss (line 5) | class BlockNeRFLoss(nn.Module):
    method __init__ (line 6) | def __init__(self, lambda_mu=0.01, Visi_loss=1e-2):
    method forward (line 11) | def forward(self, inputs, targets):
  class InterPosEmbedding (line 25) | class InterPosEmbedding(nn.Module):
    method __init__ (line 26) | def __init__(self, N_freqs=10):
    method forward (line 36) | def forward(self, mu, diagE):
  class PosEmbedding (line 57) | class PosEmbedding(nn.Module):
    method __init__ (line 58) | def __init__(self, N_freqs):
    method forward (line 69) | def forward(self, x):
  class Block_NeRF (line 77) | class Block_NeRF(nn.Module):
    method __init__ (line 78) | def __init__(self, D=8, W=256, skips=[4],
    method forward (line 124) | def forward(self, x, sigma_only=False):
  class Visibility (line 156) | class Visibility(nn.Module):
    method __init__ (line 157) | def __init__(self,
    method forward (line 172) | def forward(self, x):

FILE: block_nerf/block_visualize.py
  function visualize_depth (line 6) | def visualize_depth(depth, cmap=cv2.COLORMAP_JET):

FILE: block_nerf/learning_utils.py
  class GradualWarmupScheduler (line 12) | class GradualWarmupScheduler(_LRScheduler):
    method __init__ (line 22) | def __init__(self, optimizer, multiplier, total_epoch, after_scheduler...
    method get_lr (line 31) | def get_lr(self):
    method step_ReduceLROnPlateau (line 42) | def step_ReduceLROnPlateau(self, metrics, epoch=None):
    method step (line 56) | def step(self, epoch=None, metrics=None):
  function get_parameters (line 69) | def get_parameters(models):
  function get_optimizer (line 83) | def get_optimizer(hparams, models):
  function get_scheduler (line 103) | def get_scheduler(hparams, optimizer):
  function get_learning_rate (line 122) | def get_learning_rate(optimizer):
  function extract_model_state_dict (line 126) | def extract_model_state_dict(ckpt_path, model_name='model', prefixes_to_...
  function load_ckpt (line 143) | def load_ckpt(model, ckpt_path, model_name='model', prefixes_to_ignore=[]):

FILE: block_nerf/metrics.py
  function mse (line 4) | def mse(image_pred, image_gt, valid_mask=None, reduction='mean'):
  function psnr (line 12) | def psnr(image_pred, image_gt, valid_mask=None, reduction='mean'):
  function ssim (line 15) | def ssim(image_pred, image_gt, reduction='mean'):

FILE: block_nerf/rendering.py
  function get_cone_mean_conv (line 8) | def get_cone_mean_conv(
  function sample_pdf (line 42) | def sample_pdf(
  function volume_rendering (line 91) | def volume_rendering(
  function render_rays (line 129) | def render_rays(

FILE: block_nerf/waymo_dataset.py
  function get_ray_directions (line 13) | def get_ray_directions(H, W, K):
  function get_rays (line 27) | def get_rays(directions, c2w):
  function find_idx_name (line 39) | def find_idx_name(elements, img_name):
  function find_nearest_idx (line 46) | def find_nearest_idx(img_source,block_elements,train_meta):
  class WaymoDataset (line 62) | class WaymoDataset(Dataset):
    method __init__ (line 63) | def __init__(self, root_dir, split='train', block='block_0',
    method read_json (line 83) | def read_json(self):
    method __len__ (line 197) | def __len__(self):
    method __getitem__ (line 208) | def __getitem__(self, idx):

FILE: data_preprocess/comvog_organize_by_cam_pos.py
  function get_pix2cam (line 17) | def get_pix2cam(focals, width, height):
  function reorder_meta (line 48) | def reorder_meta(meta_dict):
  function form_unified_dict (line 69) | def form_unified_dict(old_to_new, metas, save_prefix='images_train', spl...

FILE: data_preprocess/fetch_data_from_tf_record.py
  function test_rays_dir_radii (line 18) | def test_rays_dir_radii(ray_dirs):
  function decode_fn (line 28) | def decode_fn(record_bytes):
  function get_cam_rays (line 45) | def get_cam_rays(H, W, K):
  function get_rotate_one_image (line 57) | def get_rotate_one_image(cam_ray_dir, world_ray_dir):
  function handle_one_record (line 80) | def handle_one_record(tfrecord, train_index, val_index):

FILE: data_preprocess/get_one_block_meta_and_visualize.py
  function get_pix2cam (line 12) | def get_pix2cam(focals, width, height):
  function form_unified_dict (line 53) | def form_unified_dict(images, metas, save_prefix='images_train', split_p...

FILE: data_preprocess/split_block.py
  function get_hparams (line 15) | def get_hparams():
  function extract_imgname_origins (line 30) | def extract_imgname_origins(meta):
  function resort_origins (line 40) | def resort_origins(img_origins, positions):
  function get_the_distance (line 54) | def get_the_distance(r=2, overlap=0.5):
  function get_each_block_element_train (line 76) | def get_each_block_element_train(img_train_origins, centroid, radius):
  function extract_img_base_camidx (line 89) | def extract_img_base_camidx(cam_idx, train_meta):
  function get_block_idx (line 98) | def get_block_idx(img_name, split_train_results):
  function get_val_block_index (line 107) | def get_val_block_index(img_val_origins, train_meta, val_meta, img_train...
  function split_dataset (line 137) | def split_dataset(train_meta, val_meta, radius=0.5, overlap=0.5):
  function extract_origins (line 195) | def extract_origins(meta):
  function extract_cam_idx (line 208) | def extract_cam_idx(train_meta):
  function extract_img_each_idx (line 217) | def extract_img_each_idx(idx,train_meta,train_split_meta):
  function transfer_pt_to_json (line 227) | def transfer_pt_to_json(pt_meta):

FILE: data_preprocess/vis.py
  function normalize (line 13) | def normalize(v):
  function vis_lr (line 18) | def vis_lr(lr_init: float = 5e-4,
  function save_image_tensor (line 47) | def save_image_tensor(image: torch.tensor, height: int, width: int, save...
  function save_images (line 67) | def save_images(rgb, dist, acc, path, idx):
  function visualize_depth (line 76) | def visualize_depth(depth, cmap=cv2.COLORMAP_JET):
  function gen_render_path (line 93) | def gen_render_path(c2ws, N_views=30):
  function create_spiral_poses (line 125) | def create_spiral_poses(radii, focus_depth, n_poses=120):
  function create_spheric_poses (line 160) | def create_spheric_poses(radius, n_poses=120):
  function stack_rgb (line 202) | def stack_rgb(rgb_gt, coarse_rgb, fine_rgb):

FILE: data_preprocess/visualize_cameras.py
  function get_camera_frustum (line 14) | def get_camera_frustum(img_size, K, W2C, frustum_length=0.5, color=[0.0,...
  function frustums2lineset (line 51) | def frustums2lineset(frustums):
  function visualize_cameras (line 70) | def visualize_cameras(
  function convert_pose (line 118) | def convert_pose(C2W):
  function read_single_scale_cam (line 126) | def read_single_scale_cam(path):
  function read_multi_scale_cam (line 158) | def read_multi_scale_cam(path):
  function create_spheric_cam (line 217) | def create_spheric_cam(focal, img_size, radius, camera_size=1):

FILE: docs/parse_markdown.py
  class MyHTMLParser (line 9) | class MyHTMLParser(HTMLParser):
    method __init__ (line 10) | def __init__(self, *, convert_charrefs: bool = ...) -> None:
    method handle_starttag (line 14) | def handle_starttag(self, tag, attrs):
    method handle_endtag (line 25) | def handle_endtag(self, tag):
    method handle_data (line 29) | def handle_data(self, data):

FILE: docs/render_docs_from_csv.py
  function check_nan (line 10) | def check_nan(item):
  function title_by_cls (line 16) | def title_by_cls(cls_str, cn):
  function write_item_to_md (line 23) | def write_item_to_md(md, data, idx, cn=False):
  function render_main_doc (line 46) | def render_main_doc(meta_data_path="docs/weekly_nerf_meta_data.xlsx", cn...

FILE: eval_block_nerf.py
  function get_hparams (line 17) | def get_hparams():
  function batched_inference (line 45) | def batched_inference(
  function filter_cam_info_by_index (line 78) | def filter_cam_info_by_index(index, cam_infos):
  function filter_Block (line 86) | def filter_Block(begin, blocks):
  function DistanceWeight (line 95) | def DistanceWeight(point, centroid, p=4):
  function Inverse_Interpolation (line 101) | def Inverse_Interpolation(model_result, W_H):

FILE: run_FourierGrid.py
  function config_parser (line 15) | def config_parser():
  function seed_everything (line 69) | def seed_everything():

FILE: train_block_nerf.py
  function get_opts (line 19) | def get_opts():
  function main (line 101) | def main(hparams):