SYMBOL INDEX (105 symbols across 17 files)

FILE: cfg.py
  class Config (line 6) | class Config:
    method __init__ (line 7) | def __init__(self, config_file):

FILE: data_generation/habitat_renderer.py
  function init_habitat (line 27) | def init_habitat(config) :
  function save_renders (line 117) | def save_renders(save_path, observation, enable_semantic, suffix=""):
  function render (line 143) | def render(sim, config):
  function set_agent_position (line 189) | def set_agent_position(sim, pose):
  function main (line 202) | def main():

FILE: data_generation/settings.py
  function make_cfg (line 48) | def make_cfg(settings):

FILE: data_generation/transformation.py
  function habitat_world_transformations (line 5) | def habitat_world_transformations():
  function opencv_to_opengl_camera (line 18) | def opencv_to_opengl_camera(transform=None):
  function opengl_to_opencv_camera (line 25) | def opengl_to_opencv_camera(transform=None):
  function Twc_to_Thc (line 32) | def Twc_to_Thc(T_wc):  # opencv-camera to world transformation ---> habi...
  function Thc_to_Twc (line 38) | def Thc_to_Twc(T_hc):  # habitat-caemra to habitat world transformation ...
  function combine_pose (line 44) | def combine_pose(t: np.array, q: quaternion.quaternion) -> np.array:

FILE: dataset.py
  function next_live_data (line 14) | def next_live_data(track_to_map_IDT, inited):
  function init_loader (line 39) | def init_loader(cfg, multi_worker=True):
  class Replica (line 63) | class Replica(Dataset):
    method __init__ (line 64) | def __init__(self, cfg):
    method __len__ (line 78) | def __len__(self):
    method __getitem__ (line 81) | def __getitem__(self, idx):
  class ScanNet (line 150) | class ScanNet(Dataset):
    method __init__ (line 151) | def __init__(self, cfg):
    method load_poses (line 191) | def load_poses(self, path):
    method __len__ (line 205) | def __len__(self):
    method __getitem__ (line 208) | def __getitem__(self, index):

FILE: embedding.py
  function positional_encoding (line 4) | def positional_encoding(
  class UniDirsEmbed (line 43) | class UniDirsEmbed(torch.nn.Module):
    method __init__ (line 44) | def __init__(self, min_deg=0, max_deg=2, scale=2.):
    method forward (line 82) | def forward(self, x):

FILE: image_transforms.py
  class BGRtoRGB (line 5) | class BGRtoRGB(object):
    method __call__ (line 8) | def __call__(self, image):
  class DepthScale (line 13) | class DepthScale(object):
    method __init__ (line 16) | def __init__(self, scale):
    method __call__ (line 19) | def __call__(self, depth):
  class DepthFilter (line 24) | class DepthFilter(object):
    method __init__ (line 27) | def __init__(self, max_depth):
    method __call__ (line 30) | def __call__(self, depth):
  class Undistort (line 36) | class Undistort(object):
    method __init__ (line 39) | def __init__(self,
    method __call__ (line 58) | def __call__(self, im):

FILE: loss.py
  function step_batch_loss (line 5) | def step_batch_loss(alpha, color, gt_depth, gt_color, sem_labels, mask_d...

FILE: metric/eval_3D_obj.py
  function calc_3d_metric (line 8) | def calc_3d_metric(mesh_rec, mesh_gt, N=200000):
  function get_gt_bg_mesh (line 43) | def get_gt_bg_mesh(gt_dir, background_cls_list):
  function get_obj_ids (line 57) | def get_obj_ids(obj_dir):

FILE: metric/eval_3D_scene.py
  function calc_3d_metric (line 7) | def calc_3d_metric(mesh_rec, mesh_gt, N=200000):

FILE: metric/metrics.py
  function completion_ratio (line 4) | def completion_ratio(gt_points, rec_points, dist_th=0.01):
  function accuracy (line 11) | def accuracy(gt_points, rec_points):
  function completion (line 18) | def completion(gt_points, rec_points):
  function chamfer (line 25) | def chamfer(gt_points, rec_points):

FILE: model.py
  function init_weights (line 4) | def init_weights(m, init_fn=torch.nn.init.xavier_normal_):
  function fc_block (line 9) | def fc_block(in_f, out_f):
  class OccupancyMap (line 16) | class OccupancyMap(torch.nn.Module):
    method __init__ (line 17) | def __init__(
    method forward (line 54) | def forward(self, x,

FILE: render_rays.py
  function occupancy_activation (line 4) | def occupancy_activation(alpha, distances=None):
  function alpha_to_occupancy (line 10) | def alpha_to_occupancy(depths, dirs, alpha, add_last=False):
  function occupancy_to_termination (line 26) | def occupancy_to_termination(occupancy, is_batch=False):
  function render (line 47) | def render(termination, vals, dim=-1):
  function render_loss (line 53) | def render_loss(render, gt, loss="L1", normalise=False):
  function reduce_batch_loss (line 67) | def reduce_batch_loss(loss_mat, var=None, avg=True, mask=None, loss_type...
  function make_3D_grid (line 98) | def make_3D_grid(occ_range=[-1., 1.], dim=256, device="cuda:0", transfor...

FILE: trainer.py
  class Trainer (line 9) | class Trainer:
    method __init__ (line 10) | def __init__(self, cfg):
    method load_network (line 26) | def load_network(self):
    method meshing (line 35) | def meshing(self, bound, obj_center, grid_dim=256):
    method eval_points (line 77) | def eval_points(self, points, chunk_size=100000):

FILE: utils.py
  class BoundingBox (line 11) | class BoundingBox():
    method __init__ (line 12) | def __init__(self):
  function bbox_open3d2bbox (line 19) | def bbox_open3d2bbox(bbox_o3d):
  function bbox_bbox2open3d (line 26) | def bbox_bbox2open3d(bbox):
  function update_vmap (line 30) | def update_vmap(models, optimiser):
  function enlarge_bbox (line 36) | def enlarge_bbox(bbox, scale, w, h):
  function get_bbox2d (line 59) | def get_bbox2d(obj_mask, bbox_scale=1.0):
  function get_bbox2d_batch (line 75) | def get_bbox2d_batch(img):
  function get_latest_queue (line 86) | def get_latest_queue(q):
  class InstData (line 101) | class InstData:
    method __init__ (line 102) | def __init__(self):
  function box_filter (line 112) | def box_filter(masks, classes, depth, inst_dict, intrinsic_open3d, T_CW,...
  function load_matrix_from_txt (line 210) | def load_matrix_from_txt(path, shape=(4, 4)):
  function check_mask_order (line 217) | def check_mask_order(obj_masks, depth_np, obj_ids):
  function unproject_pointcloud (line 257) | def unproject_pointcloud(depth, intrinsic_open3d, T_CW):
  function check_inside_ratio (line 266) | def check_inside_ratio(pc, bbox3D):
  function track_instance (line 274) | def track_instance(masks, classes, depth, inst_list, sem_dict, intrinsic...

FILE: vis.py
  function marching_cubes (line 6) | def marching_cubes(occupancy, level=0.5):
  function trimesh_to_open3d (line 21) | def trimesh_to_open3d(src):

FILE: vmap.py
  class performance_measure (line 17) | class performance_measure:
    method __init__ (line 19) | def __init__(self, name) -> None:
    method __enter__ (line 22) | def __enter__(self):
    method __exit__ (line 25) | def __exit__(self, type, value, tb):
  function origin_dirs_W (line 31) | def origin_dirs_W(T_WC, dirs_C):
  function stratified_bins (line 45) | def stratified_bins(min_depth, max_depth, n_bins, n_rays, type=torch.flo...
  function normal_bins_sampling (line 75) | def normal_bins_sampling(depth, n_bins, n_rays, delta, device = "cuda:0"):
  class sceneObject (line 90) | class sceneObject:
    method __init__ (line 96) | def __init__(self, cfg, obj_id, rgb:torch.tensor, depth:torch.tensor, ...
    method init_obj_center (line 194) | def init_obj_center(self, intrinsic_open3d, depth, mask, t_wc):
    method append_keyframe (line 208) | def append_keyframe(self, rgb:torch.tensor, depth:torch.tensor, mask:t...
    method prune_keyframe (line 265) | def prune_keyframe(self):
    method get_bound (line 270) | def get_bound(self, intrinsic_open3d):
    method get_training_samples (line 319) | def get_training_samples(self, n_frames, n_samples, cached_rays_dir):
    method sample_3d_points (line 366) | def sample_3d_points(self, sampled_rgbs, sampled_depth, origins, dirs_w):
    method save_checkpoints (line 461) | def save_checkpoints(self, path, epoch):
    method load_checkpoints (line 478) | def load_checkpoints(self, ckpt_file):
  class cameraInfo (line 494) | class cameraInfo:
    method __init__ (line 496) | def __init__(self, cfg) -> None:
    method get_rays_dirs (line 508) | def get_rays_dirs(self, depth_type="z"):