SYMBOL INDEX (492 symbols across 64 files)

FILE: arguments/__init__.py
  class GroupParams (line 16) | class GroupParams:
  class ParamGroup (line 19) | class ParamGroup:
    method __init__ (line 20) | def __init__(self, parser: ArgumentParser, name : str, fill_none = Fal...
    method extract (line 40) | def extract(self, args):
  class ModelParams (line 47) | class ModelParams(ParamGroup):
    method __init__ (line 48) | def __init__(self, parser, sentinel=False):
    method extract (line 59) | def extract(self, args):
  class PipelineParams (line 64) | class PipelineParams(ParamGroup):
    method __init__ (line 65) | def __init__(self, parser):
  class OptimizationParams (line 71) | class OptimizationParams(ParamGroup):
    method __init__ (line 72) | def __init__(self, parser):
  function get_combined_args (line 92) | def get_combined_args(parser : ArgumentParser):

FILE: configs/config.py
  class AttrDict (line 26) | class AttrDict(dict):
    method __init__ (line 29) | def __init__(self, *args, **kwargs):
    method yaml (line 41) | def yaml(self):
    method __repr__ (line 59) | def __repr__(self):
  class Config (line 83) | class Config(AttrDict):
    method __init__ (line 87) | def __init__(self, filename=None, verbose=False):
    method load_config (line 107) | def load_config(self, filename):
    method print_config (line 136) | def print_config(self, level=0):
    method save_config (line 145) | def save_config(self, logdir):
  function rsetattr (line 151) | def rsetattr(obj, attr, val):
  function rgetattr (line 157) | def rgetattr(obj, attr, *args):
  function recursive_update (line 167) | def recursive_update(d, u):
  function recursive_update_strict (line 182) | def recursive_update_strict(d, u, stack=[]):
  function parse_cmdline_arguments (line 200) | def parse_cmdline_arguments(args):

FILE: evaluation/crop_mesh.py
  function align_gt_with_cam (line 12) | def align_gt_with_cam(pts, trans):
  function main (line 18) | def main(args):

FILE: evaluation/eval_dtu/eval.py
  function sample_single_tri (line 10) | def sample_single_tri(input_):
  function write_vis_pcd (line 21) | def write_vis_pcd(file, points, colors):

FILE: evaluation/eval_dtu/evaluate_single_scene.py
  function cull_scan (line 22) | def cull_scan(scan, mesh_path, result_mesh_file, instance_dir):

FILE: evaluation/eval_dtu/render_utils.py
  function get_psnr (line 9) | def get_psnr(img1, img2, normalize_rgb=False):
  function load_rgb (line 20) | def load_rgb(path, normalize_rgb = False):
  function load_K_Rt_from_P (line 31) | def load_K_Rt_from_P(filename, P=None):
  function get_camera_params (line 55) | def get_camera_params(uv, pose, intrinsics):
  function get_camera_for_plot (line 85) | def get_camera_for_plot(pose):
  function lift (line 96) | def lift(x, y, z, intrinsics):
  function quat_to_rot (line 112) | def quat_to_rot(q):
  function rot_to_quat (line 132) | def rot_to_quat(R):
  function get_sphere_intersections (line 153) | def get_sphere_intersections(cam_loc, ray_directions, r = 1.0):

FILE: evaluation/eval_tnt.py
  function nn_correspondance (line 9) | def nn_correspondance(verts1, verts2):
  function evaluate (line 22) | def evaluate(mesh_pred, mesh_trgt, threshold=.05, down_sample=.02):
  function main (line 52) | def main(args):

FILE: evaluation/lpipsPyTorch/__init__.py
  function lpips (line 6) | def lpips(x: torch.Tensor,

FILE: evaluation/lpipsPyTorch/modules/lpips.py
  class LPIPS (line 8) | class LPIPS(nn.Module):
    method __init__ (line 17) | def __init__(self, net_type: str = 'alex', version: str = '0.1'):
    method forward (line 30) | def forward(self, x: torch.Tensor, y: torch.Tensor):

FILE: evaluation/lpipsPyTorch/modules/networks.py
  function get_network (line 12) | def get_network(net_type: str):
  class LinLayers (line 23) | class LinLayers(nn.ModuleList):
    method __init__ (line 24) | def __init__(self, n_channels_list: Sequence[int]):
  class BaseNet (line 36) | class BaseNet(nn.Module):
    method __init__ (line 37) | def __init__(self):
    method set_requires_grad (line 46) | def set_requires_grad(self, state: bool):
    method z_score (line 50) | def z_score(self, x: torch.Tensor):
    method forward (line 53) | def forward(self, x: torch.Tensor):
  class SqueezeNet (line 66) | class SqueezeNet(BaseNet):
    method __init__ (line 67) | def __init__(self):
  class AlexNet (line 77) | class AlexNet(BaseNet):
    method __init__ (line 78) | def __init__(self):
  class VGG16 (line 88) | class VGG16(BaseNet):
    method __init__ (line 89) | def __init__(self):

FILE: evaluation/lpipsPyTorch/modules/utils.py
  function normalize_activation (line 6) | def normalize_activation(x, eps=1e-10):
  function get_state_dict (line 11) | def get_state_dict(net_type: str = 'alex', version: str = '0.1'):

FILE: evaluation/metrics.py
  function readImages (line 30) | def readImages(renders_dir, gt_dir):
  function evaluate (line 42) | def evaluate(model_paths):

FILE: evaluation/render.py
  function render_set (line 28) | def render_set(model_path, name, iteration, views, gaussians, cfg, backg...
  function render_sets (line 58) | def render_sets(cfg, iteration : int, skip_train : bool, skip_test : bool):

FILE: evaluation/tnt_eval/evaluation.py
  function read_alignment_transformation (line 44) | def read_alignment_transformation(filename):
  function write_color_distances (line 50) | def write_color_distances(path, pcd, distances, max_distance):
  function EvaluateHisto (line 60) | def EvaluateHisto(
  function get_f1_score_histo2 (line 163) | def get_f1_score_histo2(threshold,

FILE: evaluation/tnt_eval/plot.py
  function plot_graph (line 40) | def plot_graph(

FILE: evaluation/tnt_eval/registration.py
  function read_mapping (line 44) | def read_mapping(filename):
  function gen_sparse_trajectory (line 58) | def gen_sparse_trajectory(mapping, f_trajectory):
  function trajectory_alignment (line 65) | def trajectory_alignment(map_file, traj_to_register, gt_traj_col, gt_trans,
  function crop_and_downsample (line 106) | def crop_and_downsample(
  function registration_unif (line 127) | def registration_unif(
  function registration_vol_ds (line 159) | def registration_vol_ds(
  function crop_based_target (line 201) | def crop_based_target(s, t):

FILE: evaluation/tnt_eval/run.py
  function run_evaluation (line 58) | def run_evaluation(dataset_dir, traj_path, ply_path, out_dir):

FILE: evaluation/tnt_eval/trajectory_io.py
  class CameraPose (line 5) | class CameraPose:
    method __init__ (line 7) | def __init__(self, meta, mat):
    method __str__ (line 11) | def __str__(self):
  function convert_trajectory_to_pointcloud (line 16) | def convert_trajectory_to_pointcloud(traj):
  function read_trajectory (line 23) | def read_trajectory(filename):
  function write_trajectory (line 38) | def write_trajectory(traj, filename):

FILE: evaluation/tnt_eval/util.py
  function make_dir (line 4) | def make_dir(path):

FILE: gaussian_renderer/__init__.py
  function render (line 22) | def render(viewpoint_camera, pc : GaussianModel, cfg, bg_color : torch.T...
  function render_fast (line 167) | def render_fast(viewpoint_camera, pc : GaussianModel, cfg, bg_color : to...
  function count_render (line 250) | def count_render(
  function visi_render (line 358) | def visi_render(
  function visi_acc_render (line 467) | def visi_acc_render(

FILE: gaussian_renderer/network_gui.py
  function init (line 26) | def init(wish_host, wish_port):
  function try_connect (line 34) | def try_connect():
  function read (line 43) | def read():
  function send (line 50) | def send(message_bytes, verify):
  function receive (line 57) | def receive():

FILE: process_data/convert.py
  function create_init_files (line 23) | def create_init_files(pinhole_dict_file, db_file, out_dir):
  function main (line 83) | def main(args):

FILE: process_data/convert_360_to_json.py
  function create_init_files (line 18) | def create_init_files(pinhole_dict_file, db_file, out_dir):
  function convert_cam_dict_to_pinhole_dict (line 76) | def convert_cam_dict_to_pinhole_dict(cam_dict, pinhole_dict_file):
  function load_COLMAP_poses (line 105) | def load_COLMAP_poses(cam_file, img_dir, tf='w2c'):
  function load_transformation (line 135) | def load_transformation(trans_file):
  function align_gt_with_cam (line 147) | def align_gt_with_cam(pts, trans):
  function main (line 153) | def main(args):

FILE: process_data/convert_data_to_json.py
  function find_closest_point (line 27) | def find_closest_point(p1, d1, p2, d2):
  function bound_by_pose (line 49) | def bound_by_pose(images):
  function bound_by_points (line 80) | def bound_by_points(points3D):
  function compute_oriented_bound (line 97) | def compute_oriented_bound(pts):
  function split_data (line 105) | def split_data(names, split=10):
  function get_split_dict (line 120) | def get_split_dict(scene_path):
  function check_concentric (line 143) | def check_concentric(images, ang_tol=np.pi / 6.0, radii_tol=0.5, pose_to...
  function export_to_json (line 177) | def export_to_json(trans, scale, scene_path, file_name, split_dict=None,...
  function data_to_json (line 197) | def data_to_json(args):

FILE: process_data/convert_dtu_to_json.py
  function load_K_Rt_from_P (line 37) | def load_K_Rt_from_P(filename, P=None):
  function dtu_to_json (line 62) | def dtu_to_json(args):
  function load_poses (line 107) | def load_poses(scene_path):
  function convert_cam_dict_to_pinhole_dict (line 128) | def convert_cam_dict_to_pinhole_dict(scene_path, pinhole_dict_file):
  function create_init_files (line 161) | def create_init_files(pinhole_dict_file, db_file, out_dir):
  function init_colmap (line 221) | def init_colmap(args):

FILE: process_data/convert_tnt_to_json.py
  function create_init_files (line 18) | def create_init_files(pinhole_dict_file, db_file, out_dir):
  function convert_cam_dict_to_pinhole_dict (line 76) | def convert_cam_dict_to_pinhole_dict(cam_dict, pinhole_dict_file):
  function load_COLMAP_poses (line 105) | def load_COLMAP_poses(cam_file, img_dir, tf='w2c'):
  function load_transformation (line 135) | def load_transformation(trans_file):
  function align_gt_with_cam (line 147) | def align_gt_with_cam(pts, trans):
  function compute_bound (line 153) | def compute_bound(pts):
  function init_colmap (line 164) | def init_colmap(args):

FILE: process_data/extract_mask.py
  function load_image (line 33) | def load_image(image_path):
  function print_ (line 48) | def print_(a):
  function load_model (line 52) | def load_model(model_config_path, model_checkpoint_path, device):
  function get_grounding_output (line 63) | def get_grounding_output(model, image, caption, box_threshold, text_thre...
  function show_mask (line 99) | def show_mask(mask, ax, random_color=False):
  function show_box (line 109) | def show_box(box, ax, label):
  function save_mask_data (line 116) | def save_mask_data(output_dir, mask_list, box_list, label_list, name):
  function morphology_open (line 132) | def morphology_open(x, k1=21, k2=21):
  function process_image (line 140) | def process_image(image_name):

FILE: process_data/extract_normal.py
  function get_args (line 45) | def get_args(test=False):
  function focal2fov (line 92) | def focal2fov(focal, pixels):
  function read_next_bytes (line 96) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact...
  function read_intrinsics_binary (line 108) | def read_intrinsics_binary(path_to_model_file):
  function read_intrinsics_text (line 137) | def read_intrinsics_text(path):
  function load_intrinsic_colmap (line 162) | def load_intrinsic_colmap(path):
  function test_samples (line 208) | def test_samples(args, model, intrins=None, device='cpu'):

FILE: python_scripts/run_base.py
  function worker (line 6) | def worker(gpu, scene, factor, fn):
  function dispatch_jobs (line 13) | def dispatch_jobs(jobs, executor, excluded_gpus, fn):
  function check_finish (line 46) | def check_finish(scene, path, type='mesh'):

FILE: python_scripts/run_dtu.py
  function train_scene (line 58) | def train_scene(gpu, scene, factor):

FILE: python_scripts/run_mipnerf360.py
  function train_scene (line 61) | def train_scene(gpu, scene, factor):

FILE: python_scripts/run_tnt.py
  function train_scene (line 61) | def train_scene(gpu, scene, factor):

FILE: python_scripts/show_360.py
  function show_matrix (line 16) | def show_matrix(scenes, output_dirs, TRIAL_NAME):

FILE: python_scripts/show_dtu.py
  function show_matrix_old (line 10) | def show_matrix_old(scenes, output_dirs, TRIAL_NAME):
  function show_matrix (line 38) | def show_matrix(scenes, output_dirs, TRIAL_NAME):

FILE: python_scripts/show_tnt.py
  function show_matrix (line 17) | def show_matrix(scenes, output_dirs, TRIAL_NAME):

FILE: scene/__init__.py
  class Scene (line 24) | class Scene:
    method __init__ (line 28) | def __init__(self, args : ModelParams, gaussians : GaussianModel, load...
    method save (line 105) | def save(self, iteration, visi=None, surf=None, save_splat=False):
    method getTrainCameras (line 119) | def getTrainCameras(self, scale=1.0):
    method getTestCameras (line 122) | def getTestCameras(self, scale=1.0):
    method getFullCameras (line 125) | def getFullCameras(self, scale=1.0):
    method getUpCameras (line 131) | def getUpCameras(self):
    method getAroundCameras (line 134) | def getAroundCameras(self):
    method getRandCameras (line 137) | def getRandCameras(self, n, up=False, around=True, sample_mode='unifor...

FILE: scene/appearance_network.py
  class UpsampleBlock (line 5) | class UpsampleBlock(nn.Module):
    method __init__ (line 6) | def __init__(self, num_input_channels, num_output_channels):
    method forward (line 12) | def forward(self, x):
  class AppearanceNetwork (line 19) | class AppearanceNetwork(nn.Module):
    method __init__ (line 20) | def __init__(self, num_input_channels, num_output_channels):
    method forward (line 34) | def forward(self, x):

FILE: scene/cameras.py
  class Camera (line 19) | class Camera(nn.Module):
    method __init__ (line 20) | def __init__(self, colmap_id, R, T, FoVx, FoVy, image, gt_alpha_mask,
  class MiniCam (line 76) | class MiniCam:
    method __init__ (line 77) | def __init__(self, width, height, fovy, fovx, znear, zfar, world_view_...
  class SampleCam (line 90) | class SampleCam(nn.Module):
    method __init__ (line 91) | def __init__(self, w2c, width, height, FoVx, FoVy, device='cuda'):
  class MiniCam2 (line 115) | class MiniCam2:
    method __init__ (line 116) | def __init__(self, c2w, width, height, fovy, fovx, znear, zfar):

FILE: scene/colmap_loader.py
  function qvec2rotmat (line 43) | def qvec2rotmat(qvec):
  function rotmat2qvec (line 55) | def rotmat2qvec(R):
  class Image (line 68) | class Image(BaseImage):
    method qvec2rotmat (line 69) | def qvec2rotmat(self):
  function read_next_bytes (line 72) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact...
  function read_points3D_text (line 83) | def read_points3D_text(path):
  function read_points3D_binary (line 125) | def read_points3D_binary(path_to_model_file):
  function read_intrinsics_text (line 156) | def read_intrinsics_text(path):
  function read_extrinsics_binary (line 180) | def read_extrinsics_binary(path_to_model_file):
  function read_intrinsics_binary (line 215) | def read_intrinsics_binary(path_to_model_file):
  function read_extrinsics_text (line 244) | def read_extrinsics_text(path):
  function read_colmap_bin_array (line 273) | def read_colmap_bin_array(path):

FILE: scene/dataset_readers.py
  class CameraInfo (line 32) | class CameraInfo(NamedTuple):
  class SceneInfo (line 47) | class SceneInfo(NamedTuple):
  function getNerfppNorm (line 57) | def getNerfppNorm(cam_info):
  function readColmapCameras (line 80) | def readColmapCameras(cam_extrinsics, cam_intrinsics, images_folder, loa...
  function fetchPly (line 149) | def fetchPly(path):
  function storePly (line 157) | def storePly(path, xyz, rgb, normals=None):
  function get_inside_mask (line 174) | def get_inside_mask(pts, trans, scale):
  function filter_point_cloud (line 180) | def filter_point_cloud(trans, scale, xyz, rgb, nb_points=5, radius=0.1):
  function readColmapSceneInfo (line 201) | def readColmapSceneInfo(path, images, eval, llffhold=8, ratio=0, split=F...
  function readCamerasFromTransforms (line 300) | def readCamerasFromTransforms(path, transformsfile, white_background, ex...
  function readNerfSyntheticInfo (line 342) | def readNerfSyntheticInfo(path, white_background, eval, extension=".png"):

FILE: scene/gaussian_model.py
  class GaussianModel (line 36) | class GaussianModel:
    method setup_functions (line 37) | def setup_functions(self):
    method __init__ (line 54) | def __init__(self, cfg):
    method capture (line 88) | def capture(self):
    method restore (line 105) | def restore(self, model_args, training_args):
    method get_scaling (line 126) | def get_scaling(self):
    method get_rotation (line 131) | def get_rotation(self):
    method get_xyz (line 135) | def get_xyz(self):
    method get_features (line 139) | def get_features(self):
    method get_objects (line 145) | def get_objects(self):
    method get_cls (line 148) | def get_cls(self, idx=None):
    method logits_2_label (line 154) | def logits_2_label(self, logits):
    method logits2prob (line 157) | def logits2prob(self, logits):
    method get_opacity (line 161) | def get_opacity(self):
    method get_apperance_embedding (line 164) | def get_apperance_embedding(self, idx):
    method get_normal (line 168) | def get_normal(self, valid=None, idx=None, refine_sign=True, is_all=Fa...
    method get_covariance (line 194) | def get_covariance(self, scaling_modifier = 1):
    method oneupSHdegree (line 197) | def oneupSHdegree(self):
    method create_from_pcd (line 201) | def create_from_pcd(self, pcd : BasicPointCloud, spatial_lr_scale : fl...
    method training_setup (line 232) | def training_setup(self, training_args, neural_sdf_params=None):
    method update_learning_rate (line 264) | def update_learning_rate(self, iteration):
    method construct_list_of_attributes (line 272) | def construct_list_of_attributes(self):
    method save_ply (line 289) | def save_ply(self, path):
    method save_inside_ply (line 323) | def save_inside_ply(self, path, inside=None):
    method save_visi_ply (line 355) | def save_visi_ply(self, path, visi):
    method reset_opacity (line 361) | def reset_opacity(self):
    method load_ply (line 366) | def load_ply(self, path):
    method replace_tensor_to_optimizer (line 425) | def replace_tensor_to_optimizer(self, tensor, name):
    method _prune_optimizer (line 442) | def _prune_optimizer(self, mask):
    method prune_points (line 462) | def prune_points(self, mask):
    method cat_tensors_to_optimizer (line 480) | def cat_tensors_to_optimizer(self, tensors_dict):
    method densification_postfix (line 504) | def densification_postfix(self, new_xyz, new_features_dc, new_features...
    method densify_and_split (line 533) | def densify_and_split(self, grads, grad_threshold, scene_extent, visi=...
    method get_dir_max_scaling (line 569) | def get_dir_max_scaling(self, scaling, rots):
    method densify_and_split_along_maxscaling (line 579) | def densify_and_split_along_maxscaling(self, grads, grad_threshold, sc...
    method densify_and_clone (line 629) | def densify_and_clone(self, grads, grad_threshold, scene_extent):
    method densify_and_prune (line 645) | def densify_and_prune(self, max_grad, min_opacity, extent, max_screen_...
    method prune_gaussians (line 661) | def prune_gaussians(self, percent, import_score: list):
    method add_densification_stats (line 669) | def add_densification_stats(self, viewspace_point_tensor, update_filter):
    method get_inside_gaus_normalized (line 673) | def get_inside_gaus_normalized(self):
    method normalize_pts (line 677) | def normalize_pts(self, pts):
    method filter_points (line 681) | def filter_points(self, nb_points=5, radius=0.01, std_ratio=0.01):
    method prune_outside (line 689) | def prune_outside(self):
    method prune_outliers (line 693) | def prune_outliers(self):
    method prune_semantics (line 699) | def prune_semantics(self, cls=BACKGROUND):

FILE: tools/camera.py
  class Pose (line 17) | class Pose():
    method __call__ (line 23) | def __call__(self, R=None, t=None):
    method invert (line 46) | def invert(self, pose, use_inverse=False):
    method compose (line 54) | def compose(self, pose_list):
    method compose_pair (line 62) | def compose_pair(self, pose_a, pose_b):
    method scale_center (line 70) | def scale_center(self, pose, scale):
    method interpolate (line 79) | def interpolate(self, pose_a, pose_b, alpha):
  class Lie (line 99) | class Lie():
    method so3_to_SO3 (line 104) | def so3_to_SO3(self, w):  # [..., 3]
    method SO3_to_so3 (line 113) | def SO3_to_so3(self, R, eps=1e-7):  # [..., 3, 3]
    method se3_to_SE3 (line 122) | def se3_to_SE3(self, wu):  # [...,3]
    method SE3_to_se3 (line 135) | def SE3_to_se3(self, Rt, eps=1e-8):  # [...,3,4]
    method skew_symmetric (line 148) | def skew_symmetric(self, w):
    method taylor_A (line 156) | def taylor_A(self, x, nth=10):
    method taylor_B (line 166) | def taylor_B(self, x, nth=10):
    method taylor_C (line 175) | def taylor_C(self, x, nth=10):
  class Quaternion (line 185) | class Quaternion():
    method q_to_R (line 187) | def q_to_R(self, q):  # [...,4]
    method R_to_q (line 197) | def R_to_q(self, R, eps=1e-6):  # [...,3,3]
    method invert (line 212) | def invert(self, q):  # [...,4]
    method product (line 218) | def product(self, q1, q2):  # [...,4]
    method interpolate (line 227) | def interpolate(self, q1, q2, alpha):  # [...,4],[...,4],[...,1]
  function to_hom (line 242) | def to_hom(X):
  function world2cam (line 249) | def world2cam(X, pose):  # [B,N,3]
  function cam2img (line 254) | def cam2img(X, cam_intr):
  function img2cam (line 258) | def img2cam(X, cam_intr):
  function cam2world (line 262) | def cam2world(X, pose):
  function angle_to_rotation_matrix (line 268) | def angle_to_rotation_matrix(a, axis):
  function get_center_and_ray (line 282) | def get_center_and_ray(pose, intr, image_size):
  function get_3D_points_from_dist (line 313) | def get_3D_points_from_dist(center, ray_unit, dist, multi=True):
  function convert_NDC (line 322) | def convert_NDC(center, ray, intr, near=1):
  function convert_NDC2 (line 342) | def convert_NDC2(center, ray, intr):
  function rotation_distance (line 363) | def rotation_distance(R1, R2, eps=1e-7):
  function get_oscil_novel_view_poses (line 371) | def get_oscil_novel_view_poses(N=60, angle=0.05, dist=5):
  function cross_product_matrix (line 382) | def cross_product_matrix(x):
  function essential_matrix (line 395) | def essential_matrix(poses):
  function fundamental_matrix (line 407) | def fundamental_matrix(poses, intr1, intr2):
  function get_ray_depth_plane_intersection (line 418) | def get_ray_depth_plane_intersection(center, ray, depths):
  function unit_view_vector_to_rotation_matrix (line 437) | def unit_view_vector_to_rotation_matrix(v, axes="ZYZ"):
  function sample_on_spherical_cap (line 456) | def sample_on_spherical_cap(anchor, N, max_angle):
  function sample_on_spherical_cap_northern (line 486) | def sample_on_spherical_cap_northern(anchor, N, max_angle, away_from=Non...

FILE: tools/camera_utils.py
  function loadCam (line 30) | def loadCam(args, id, cam_info, resolution_scale):
  function cameraList_from_camInfos (line 79) | def cameraList_from_camInfos(cam_infos, resolution_scale, args):
  function camera_to_JSON (line 88) | def camera_to_JSON(id, camera):
  function find_up_axis (line 111) | def find_up_axis(R):
  function find_axis (line 124) | def find_axis(R, axis_name='up'):
  function dot (line 144) | def dot(x, y):
  function length (line 151) | def length(x, eps=1e-20):
  function safe_normalize (line 158) | def safe_normalize(x, eps=1e-20):
  function look_at_np (line 162) | def look_at_np(campos, target, opengl=True):
  function look_at (line 182) | def look_at(campos, target, opengl=True):
  function orbit_camera (line 203) | def orbit_camera(elevation, azimuth, radius=1, is_degree=True, target=No...
  function cubic_camera (line 223) | def cubic_camera(n, trans, scale, target=None, opengl=False):
  function check_tensor (line 243) | def check_tensor(x):
  function up_camera (line 249) | def up_camera(n, trans, scale, target=None, opengl=False): # colmap
  function around_camera (line 278) | def around_camera(n, trans, scale, height=None, target=None, opengl=False):
  function bb_camera (line 315) | def bb_camera(n, trans, scale, height=None, target=None, opengl=False, u...
  function around_grid_posi (line 404) | def around_grid_posi(num_points, scale, right_axis, up_axis, front_axis,...
  function up_grid_posi (line 463) | def up_grid_posi(num_points, scale, right_axis, up_axis, front_axis):
  function grid_camera (line 484) | def grid_camera(trans, scale, opengl=False):
  function sample_cameras (line 519) | def sample_cameras(model, n, up=False, around=True, look_mode='target', ...
  class OrbitCamera (line 540) | class OrbitCamera:
    method __init__ (line 541) | def __init__(self, W, H, r=2, fovy=60, near=0.01, far=100):
    method fovx (line 553) | def fovx(self):
    method campos (line 557) | def campos(self):
    method pose (line 562) | def pose(self):
    method view (line 576) | def view(self):
    method perspective (line 581) | def perspective(self):
    method intrinsics (line 601) | def intrinsics(self):
    method mvp (line 606) | def mvp(self):
    method orbit (line 609) | def orbit(self, dx, dy):
    method scale (line 616) | def scale(self, delta):
    method pan (line 619) | def pan(self, dx, dy, dz=0):

FILE: tools/crop_mesh.py
  function align_gt_with_cam (line 7) | def align_gt_with_cam(pts, trans):
  function filter_largest_cc (line 13) | def filter_largest_cc(mesh):
  function main (line 23) | def main(args):

FILE: tools/denoise_pcd.py
  function remove_radius_outlier (line 4) | def remove_radius_outlier(xyz, nb_points=5, radius=0.1):
  function remove_statistical_outlier (line 12) | def remove_statistical_outlier(xyz, nb_points=20, std_ratio=20.):

FILE: tools/depth2mesh.py
  function tsdf_fusion (line 22) | def tsdf_fusion(args, cfg, model, cameras, dirs, bg, outdir, mesh_name='...
  function tsdf_cpu (line 103) | def tsdf_cpu(args, cfg, model, cameras, dirs, bg, outdir, mesh_name='fus...
  function main (line 123) | def main(args):

FILE: tools/distributed.py
  function init_dist (line 21) | def init_dist(local_rank, backend='nccl', **kwargs):
  function get_rank (line 38) | def get_rank():
  function get_world_size (line 47) | def get_world_size():
  function broadcast_object_list (line 56) | def broadcast_object_list(message, src=0):
  function master_only (line 65) | def master_only(func):
  function is_master (line 77) | def is_master():
  function is_dist (line 82) | def is_dist():
  function barrier (line 86) | def barrier():
  function master_first (line 92) | def master_first():
  function is_local_master (line 100) | def is_local_master():
  function master_only_print (line 105) | def master_only_print(*args):
  function dist_reduce_tensor (line 110) | def dist_reduce_tensor(tensor, rank=0, reduce='mean'):
  function dist_all_reduce_tensor (line 127) | def dist_all_reduce_tensor(tensor, reduce='mean'):
  function dist_all_gather_tensor (line 143) | def dist_all_gather_tensor(tensor):

FILE: tools/general_utils.py
  function inverse_sigmoid (line 22) | def inverse_sigmoid(x):
  function PILtoTorch (line 26) | def PILtoTorch(pil_image, resolution):
  function NumpytoTorch (line 35) | def NumpytoTorch(image, resolution):
  function get_expon_lr_func (line 49) | def get_expon_lr_func(
  function strip_lowerdiag (line 84) | def strip_lowerdiag(L):
  function strip_symmetric (line 95) | def strip_symmetric(sym):
  function build_rotation (line 98) | def build_rotation(r):
  function build_scaling_rotation (line 121) | def build_scaling_rotation(s, r):
  function safe_state (line 132) | def safe_state(silent):
  function set_random_seed (line 151) | def set_random_seed(seed):

FILE: tools/graphics_utils.py
  class BasicPointCloud (line 17) | class BasicPointCloud(NamedTuple):
  function geom_transform_points (line 22) | def geom_transform_points(points, transf_matrix):
  function getWorld2View (line 31) | def getWorld2View(R, t):
  function getWorld2View2 (line 38) | def getWorld2View2(R, t, translate=np.array([.0, .0, .0]), scale=1.0):
  function getView2World (line 51) | def getView2World(R, t):
  function getProjectionMatrix (line 63) | def getProjectionMatrix(znear, zfar, fovX, fovY):
  function getIntrinsic (line 89) | def getIntrinsic(fovX, fovY, h, w):
  function fov2focal (line 104) | def fov2focal(fov, pixels):
  function focal2fov (line 107) | def focal2fov(focal, pixels):
  function ndc_2_cam (line 111) | def ndc_2_cam(ndc_xyz, intrinsic, W, H):
  function depth2point_cam (line 120) | def depth2point_cam(sampled_depth, ref_intrinsic):
  function depth2point (line 134) | def depth2point(depth_image, intrinsic_matrix, extrinsic_matrix):
  function get_all_px_dir (line 144) | def get_all_px_dir(intrinsics, height, width):

FILE: tools/image_utils.py
  function mse (line 14) | def mse(img1, img2):
  function psnr (line 17) | def psnr(img1, img2):

FILE: tools/loss_utils.py
  function entropy_loss (line 30) | def entropy_loss(opacity):
  function l1_loss (line 36) | def l1_loss(network_output, gt):
  function log_l1_loss (line 40) | def log_l1_loss(network_output, gt):
  function l2_loss (line 45) | def l2_loss(network_output, gt):
  function gaussian (line 49) | def gaussian(window_size, sigma):
  function create_window (line 54) | def create_window(window_size, channel):
  function ssim (line 61) | def ssim(img1, img2, window_size=11, size_average=True):
  function _ssim (line 72) | def _ssim(img1, img2, window, window_size, channel, size_average=True):
  function eikonal_loss (line 95) | def eikonal_loss(gradients):
  function curvature_loss (line 102) | def curvature_loss(hessian):
  function compute_normal_loss (line 109) | def compute_normal_loss(normal_pred, normal_gt, weight=None):
  function monosdf_normal_loss (line 122) | def monosdf_normal_loss(normal_pred: torch.Tensor, normal_gt: torch.Tens...
  function cos_weight (line 135) | def cos_weight(render_normal, gt_normal, exp_t=1.0):
  function compute_scale_and_shift (line 146) | def compute_scale_and_shift(prediction, target, mask):
  function reduction_batch_based (line 169) | def reduction_batch_based(image_loss, M):
  function reduction_image_based (line 181) | def reduction_image_based(image_loss, M):
  function mse_loss (line 192) | def mse_loss(prediction, target, mask, reduction=reduction_batch_based):
  function gradient_loss (line 201) | def gradient_loss(prediction, target, mask, reduction=reduction_batch_ba...
  class MSELoss (line 221) | class MSELoss(nn.Module):
    method __init__ (line 222) | def __init__(self, reduction='batch-based'):
    method forward (line 230) | def forward(self, prediction, target, mask):
  class GradientLoss (line 234) | class GradientLoss(nn.Module):
    method __init__ (line 235) | def __init__(self, scales=4, reduction='batch-based'):
    method forward (line 245) | def forward(self, prediction, target, mask):
  class ScaleAndShiftInvariantLoss (line 257) | class ScaleAndShiftInvariantLoss(nn.Module):
    method __init__ (line 258) | def __init__(self, alpha=0.5, scales=1, reduction='batch-based'):
    method forward (line 267) | def forward(self, prediction, target, mask=None):
    method __get_prediction_ssi (line 280) | def __get_prediction_ssi(self):
  function normal2curv (line 287) | def normal2curv(normal, mask = None):
  function L1_loss_appearance (line 303) | def L1_loss_appearance(image, gt_image, gaussians, view_idx, return_tran...

FILE: tools/math_utils.py
  function eps_sqrt (line 4) | def eps_sqrt(squared, eps=1e-17):
  function ndc_to_pix (line 12) | def ndc_to_pix(p, resolution):
  function decompose_to_R_and_t (line 25) | def decompose_to_R_and_t(transform_mat, row_major=True):
  function to_homogen (line 39) | def to_homogen(x, dim=-1):
  function normalize_pts (line 50) | def normalize_pts(pts, trans, scale):
  function inv_normalize_pts (line 61) | def inv_normalize_pts(pts, trans, scale):
  function get_inside_normalized (line 70) | def get_inside_normalized(xyz, trans, scale):

FILE: tools/mcube_utils.py
  function marching_cubes_with_contraction (line 17) | def marching_cubes_with_contraction(

FILE: tools/mesh_utils.py
  function post_process_mesh (line 15) | def post_process_mesh(mesh, cluster_to_keep=1000):
  function to_cam_open3d (line 38) | def to_cam_open3d(viewpoint_stack):
  class GaussianExtractor (line 57) | class GaussianExtractor(object):
    method __init__ (line 58) | def __init__(self, gaussians, render, cfg, bg_color=None, dirs=None, p...
    method clean (line 78) | def clean(self):
    method reconstruction (line 87) | def reconstruction(self, viewpoint_stack):
    method extract_mesh_bounded (line 125) | def extract_mesh_bounded(self, voxel_size=0.004, sdf_trunc=0.02, depth...
    method extract_mesh_unbounded (line 169) | def extract_mesh_unbounded(self, resolution=1024):
    method export_image (line 279) | def export_image(self, path):

FILE: tools/normal_utils.py
  function get_normal_sign (line 7) | def get_normal_sign(normals, begin=None, end=None, trans=None, mode='ori...
  function compute_gradient (line 24) | def compute_gradient(img):
  function compute_normals (line 30) | def compute_normals(depth_map, K):
  function compute_edge (line 44) | def compute_edge(image, k=11, thr=0.01):
  function get_edge_aware_distortion_map (line 57) | def get_edge_aware_distortion_map(gt_image, distortion_map):

FILE: tools/prune.py
  function calculate_v_imp_score (line 6) | def calculate_v_imp_score(gaussians, imp_list, v_pow):
  function prune_list (line 25) | def prune_list(gaussians, viewpoint_stack, pipe, background):
  function get_visi_list (line 51) | def get_visi_list(gaussians, viewpoint_stack, pipe, background):

FILE: tools/render_utils.py
  function normalize (line 26) | def normalize(x: np.ndarray) -> np.ndarray:
  function pad_poses (line 30) | def pad_poses(p: np.ndarray) -> np.ndarray:
  function unpad_poses (line 36) | def unpad_poses(p: np.ndarray) -> np.ndarray:
  function recenter_poses (line 41) | def recenter_poses(poses: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
  function average_pose (line 49) | def average_pose(poses: np.ndarray) -> np.ndarray:
  function viewmatrix (line 57) | def viewmatrix(lookdir: np.ndarray, up: np.ndarray,
  function focus_point_fn (line 66) | def focus_point_fn(poses: np.ndarray) -> np.ndarray:
  function transform_poses_pca (line 74) | def transform_poses_pca(poses: np.ndarray) -> Tuple[np.ndarray, np.ndarr...
  function generate_ellipse_path (line 108) | def generate_ellipse_path(poses: np.ndarray,
  function generate_path (line 153) | def generate_path(viewpoint_cameras, n_frames=480):
  function load_img (line 176) | def load_img(pth: str) -> np.ndarray:
  function create_videos (line 183) | def create_videos(base_dir, input_dir, out_name, num_frames=480):
  function save_img_u8 (line 249) | def save_img_u8(img, pth):
  function save_img_f32 (line 257) | def save_img_f32(depthmap, pth):

FILE: tools/sh_utils.py
  function eval_sh (line 57) | def eval_sh(deg, sh, dirs):
  function RGB2SH (line 114) | def RGB2SH(rgb):
  function SH2RGB (line 117) | def SH2RGB(sh):

FILE: tools/system_utils.py
  function mkdir_p (line 16) | def mkdir_p(folder_path):
  function searchForMaxIteration (line 26) | def searchForMaxIteration(folder):

FILE: tools/termcolor.py
  function red (line 18) | def red(x): return termcolor.colored(str(x), color="red")
  function green (line 19) | def green(x): return termcolor.colored(str(x), color="green")
  function blue (line 20) | def blue(x): return termcolor.colored(str(x), color="blue")
  function cyan (line 21) | def cyan(x): return termcolor.colored(str(x), color="cyan")
  function yellow (line 22) | def yellow(x): return termcolor.colored(str(x), color="yellow")
  function magenta (line 23) | def magenta(x): return termcolor.colored(str(x), color="magenta")
  function grey (line 24) | def grey(x): return termcolor.colored(str(x), color="grey")
  function PP (line 32) | def PP(x):
  function alert (line 39) | def alert(x, color='red'):

FILE: tools/visualization.py
  function wandb_image (line 20) | def wandb_image(images, from_range=(0, 1)):
  function preprocess_image (line 26) | def preprocess_image(images, from_range=(0, 1), cmap="viridis"):
  function wandb_sem (line 36) | def wandb_sem(image, palette=PALETTE):
  function tensor2pil (line 44) | def tensor2pil(images):
  function get_heatmap (line 50) | def get_heatmap(gray, cmap):  # [N,H,W]
  function save_render (line 56) | def save_render(render, path):

FILE: tools/visualize.py
  function get_camera_mesh (line 22) | def get_camera_mesh(pose, depth=1):
  function merge_meshes (line 39) | def merge_meshes(vertices, faces):
  function merge_wireframes_k3d (line 46) | def merge_wireframes_k3d(wireframe):
  function merge_wireframes_plotly (line 52) | def merge_wireframes_plotly(wireframe):
  function get_xyz_indicators (line 58) | def get_xyz_indicators(pose, length=0.1):
  function merge_xyz_indicators_k3d (line 64) | def merge_xyz_indicators_k3d(xyz):  # [N,4,3]
  function merge_xyz_indicators_plotly (line 72) | def merge_xyz_indicators_plotly(xyz):  # [N,4,3]
  function k3d_visualize_pose (line 81) | def k3d_visualize_pose(poses, vis_depth=0.5, xyz_length=0.1, center_size...
  function plotly_visualize_pose (line 120) | def plotly_visualize_pose(poses, vis_depth=0.5, xyz_length=0.5, center_s...

FILE: train.py
  function parse_args (line 10) | def parse_args():
  function main (line 19) | def main():

FILE: trainer.py
  class Trainer (line 42) | class Trainer(object):
    method __init__ (line 43) | def __init__(self, cfg):
    method setup_model (line 68) | def setup_model(self, cfg):
    method setup_dataset (line 71) | def setup_dataset(self, cfg):
    method init_writer (line 78) | def init_writer(self, cfg):
    method init_wandb (line 98) | def init_wandb(self, cfg, wandb_id=None, project="", run_name=None, mo...
    method init_losses (line 138) | def init_losses(self):
    method setup_optimizer (line 147) | def setup_optimizer(self, cfg):
    method init_attributes (line 150) | def init_attributes(self):
    method train (line 199) | def train(self):
    method get_center_scale (line 212) | def get_center_scale(self):
    method model_forward (line 225) | def model_forward(self, data, mode):
    method _compute_loss (line 233) | def _compute_loss(self, data, mode=None):
    method _get_total_loss (line 310) | def _get_total_loss(self):
    method train_step (line 323) | def train_step(self, mode='train'):
    method start_of_iteration (line 394) | def start_of_iteration(self):
    method end_of_iteration (line 406) | def end_of_iteration(self, output, render, progress_bar):
    method log_wandb_scalars (line 438) | def log_wandb_scalars(self, output, mode=None):
    method log_wandb_images (line 452) | def log_wandb_images(self, data, mode=None):
    method log_hist (line 496) | def log_hist(self, tensor, name, num_bin=10):
    method test (line 505) | def test(self, renderFunc):
    method finalize (line 589) | def finalize(self):
    method log_writer (line 593) | def log_writer(self, mode=None):
    method save_vis (line 598) | def save_vis(self, data, name, mode='train'):
    method sample_cameras (line 621) | def sample_cameras(self, n, up=False, around=True, look_mode='target',...
    method get_visi_mask (line 637) | def get_visi_mask(self, n=500, up=False, around=True, denoise_after=Fa...
    method get_visi_mask_acc (line 688) | def get_visi_mask_acc(self, n=500, up=False, around=True, sample_mode=...
    method save_gaussians (line 705) | def save_gaussians(self):