SYMBOL INDEX (189 symbols across 30 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):
    method extract (line 111) | def extract(self, args):
  function get_combined_args (line 127) | def get_combined_args(parser : ArgumentParser):

FILE: gaussian_renderer/__init__.py
  function render (line 22) | def render(viewpoint_camera, pc : GaussianModel, pipe, bg_color : torch....

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: lpipsPyTorch/__init__.py
  function lpips (line 6) | def lpips(x: torch.Tensor,

FILE: 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: 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: 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: metrics.py
  function readImages (line 24) | def readImages(renders_dir, gt_dir):
  function evaluate (line 36) | def evaluate(model_paths):

FILE: render.py
  function render_set (line 33) | def render_set(model_path, name, iteration, views, gaussians, pipeline, ...
  function render_sets (line 88) | def render_sets(dataset : ModelParams, iteration : int, pipeline : Pipel...

FILE: render_lerf_by_text.py
  function render_set (line 33) | def render_set(model_path, name, iteration, views, gaussians, pipeline, ...
  function render_sets (line 186) | def render_sets(dataset : ModelParams, iteration : int, pipeline : Pipel...

FILE: scene/__init__.py
  class Scene (line 21) | class Scene:
    method __init__ (line 25) | def __init__(self, args : ModelParams, gaussians : GaussianModel, load...
    method save (line 86) | def save(self, iteration, save_q=[]):
    method getTrainCameras (line 90) | def getTrainCameras(self, scale=1.0):
    method getTestCameras (line 93) | def getTestCameras(self, scale=1.0):

FILE: scene/cameras.py
  class Camera (line 17) | class Camera(nn.Module):
    method __init__ (line 18) | def __init__(self, colmap_id, R, T, FoVx, FoVy, cx, cy, image, depth, ...
    method to_gpu (line 77) | def to_gpu(self):
    method to_cpu (line 85) | def to_cpu(self):
  class MiniCam (line 92) | class MiniCam:
    method __init__ (line 93) | def __init__(self, width, height, fovy, fovx, znear, zfar, world_view_...

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 28) | class CameraInfo(NamedTuple):
  class SceneInfo (line 45) | class SceneInfo(NamedTuple):
  function getNerfppNorm (line 52) | def getNerfppNorm(cam_info):
  function readColmapCameras (line 75) | def readColmapCameras(cam_extrinsics, cam_intrinsics, images_folder):
  function fetchPly (line 140) | def fetchPly(path):
  function storePly (line 155) | def storePly(path, xyz, rgb):
  function readColmapSceneInfo (line 172) | def readColmapSceneInfo(path, images, eval, llffhold=8):
  function readCamerasFromTransforms (line 219) | def readCamerasFromTransforms(path, transformsfile, white_background, ex...
  function readNerfSyntheticInfo (line 324) | def readNerfSyntheticInfo(path, white_background, eval, extension=".png"):

FILE: scene/gaussian_model.py
  function sigmoid (line 25) | def sigmoid(x):
  function distCUDA2 (line 28) | def distCUDA2(points):
  class GaussianModel (line 38) | class GaussianModel:
    method setup_functions (line 40) | def setup_functions(self):
    method __init__ (line 58) | def __init__(self, sh_degree : int):
    method capture (line 78) | def capture(self):
    method restore (line 96) | def restore(self, model_args, training_args):
    method get_scaling (line 117) | def get_scaling(self):
    method get_scaling_origin (line 121) | def get_scaling_origin(self):
    method get_rotation (line 125) | def get_rotation(self):
    method get_rotation_matrix (line 129) | def get_rotation_matrix(self):
    method get_eigenvector (line 133) | def get_eigenvector(self):
    method get_xyz (line 142) | def get_xyz(self):
    method get_features (line 146) | def get_features(self):
    method get_opacity (line 152) | def get_opacity(self):
    method get_ins_feat (line 157) | def get_ins_feat(self, origin=False):
    method get_covariance (line 165) | def get_covariance(self, scaling_modifier = 1):
    method oneupSHdegree (line 168) | def oneupSHdegree(self):
    method create_from_pcd (line 172) | def create_from_pcd(self, pcd : BasicPointCloud, spatial_lr_scale : fl...
    method training_setup (line 202) | def training_setup(self, training_args):
    method update_learning_rate (line 227) | def update_learning_rate(self, iteration, root_start, leaf_start):
    method construct_list_of_attributes (line 240) | def construct_list_of_attributes(self):
    method save_ply (line 255) | def save_ply(self, path, save_q=[]):
    method reset_opacity (line 289) | def reset_opacity(self):
    method load_ply (line 294) | def load_ply(self, path):
    method replace_tensor_to_optimizer (line 346) | def replace_tensor_to_optimizer(self, tensor, name):
    method _prune_optimizer (line 361) | def _prune_optimizer(self, mask):
    method prune_points (line 379) | def prune_points(self, mask):
    method cat_tensors_to_optimizer (line 396) | def cat_tensors_to_optimizer(self, tensors_dict):
    method densification_postfix (line 418) | def densification_postfix(self, new_xyz, new_features_dc, new_features...
    method densify_and_split (line 441) | def densify_and_split(self, grads, grad_threshold, scene_extent, N=2):
    method densify_and_clone (line 468) | def densify_and_clone(self, grads, grad_threshold, scene_extent):
    method densify_and_prune (line 485) | def densify_and_prune(self, max_grad, min_opacity, extent, max_screen_...
    method add_densification_stats (line 501) | def add_densification_stats(self, viewspace_point_tensor, update_filter):

FILE: scene/kmeans_quantize.py
  class Quantize_kMeans (line 12) | class Quantize_kMeans():
    method __init__ (line 13) | def __init__(self, num_clusters=64, num_leaf_clusters=10, num_iters=10...
    method get_dist (line 38) | def get_dist(self, x, y, mode='sq_euclidean'):
    method update_centers (line 58) | def update_centers(self, feat, mode="root", selected_leaf=-1):
    method update_centers_ (line 82) | def update_centers_(self, feat, cluster_mask=None, nn_index=None, avg=...
    method equalize_cluster_size (line 89) | def equalize_cluster_size(self, mode="root"):
    method cluster_assign (line 146) | def cluster_assign(self, feat, feat_scaled=None, mode="root", selected...
    method rescale (line 243) | def rescale(self, feat, scale=None):
    method forward (line 252) | def forward(self, gaussian, iteration, assign=False, mode="root", sele...
    method replace_with_centers (line 277) | def replace_with_centers(self, gaussian):

FILE: scripts/compute_lerf_iou.py
  function load_image_as_binary (line 6) | def load_image_as_binary(image_path, is_png=False, threshold=10):
  function calculate_iou (line 14) | def calculate_iou(mask1, mask2):
  function evalute (line 21) | def evalute(gt_base, pred_base, scene_name):

FILE: scripts/eval_scannet.py
  function sigmoid (line 29) | def sigmoid(x):
  function write_ply (line 32) | def write_ply(vertex_data, output_path):
  function read_labels_from_ply (line 47) | def read_labels_from_ply(file_path):
  function calculate_metrics (line 55) | def calculate_metrics(gt, pred, total_classes):

FILE: scripts/render_by_click.py
  function get_pixel_values (line 35) | def get_pixel_values(image_path, position, radius=10):
  function compute_click_values (line 55) | def compute_click_values(model_path, image_name, pix_xy, radius=5):
  function render_set (line 69) | def render_set(model_path, name, iteration, views, gaussians, pipeline, ...
  function render_sets (line 266) | def render_sets(dataset : ModelParams, iteration : int, pipeline : Pipel...

FILE: scripts/scannet2blender.py
  function load_transform_matrix (line 5) | def load_transform_matrix(file_path):
  function process_directory (line 13) | def process_directory(directory_path):

FILE: scripts/vis_opengs_pts_feat.py
  function sigmoid (line 5) | def sigmoid(x):
  function visualize_ply (line 9) | def visualize_ply(ply_path):

FILE: train.py
  function dec2binary (line 52) | def dec2binary(x, n_bits=None):
  function save_kmeans (line 62) | def save_kmeans(kmeans_list, quantized_params, out_dir, mode="root"):
  function cohesion_loss (line 102) | def cohesion_loss(feat_map, gt_mask, feat_mean_stack):
  function separation_loss (line 123) | def separation_loss(feat_mean_stack, iteration):
  function training (line 157) | def training(dataset, opt, pipe, testing_iterations, saving_iterations, ...
  function prepare_output_and_logger (line 596) | def prepare_output_and_logger(args):
  function construct_pseudo_ins_feat (line 618) | def construct_pseudo_ins_feat(scene : Scene, renderFunc, renderArgs,
  function training_report (line 912) | def training_report(tb_writer, iteration, Ll1, loss, l1_loss, elapsed, t...
  function initialize_new_params (line 952) | def initialize_new_params(new_pt_cld, mean3_sq_dist):

FILE: utils/camera_utils.py
  function loadCam (line 20) | def loadCam(args, id, cam_info, resolution_scale):
  function cameraList_from_camInfos (line 76) | def cameraList_from_camInfos(cam_infos, resolution_scale, args):
  function camera_to_JSON (line 84) | def camera_to_JSON(id, camera : Camera):

FILE: utils/general_utils.py
  function inverse_sigmoid (line 18) | def inverse_sigmoid(x):
  function PILtoTorch (line 21) | def PILtoTorch(pil_image, resolution):
  function get_expon_lr_func (line 29) | def get_expon_lr_func(
  function strip_lowerdiag (line 64) | def strip_lowerdiag(L):
  function strip_symmetric (line 75) | def strip_symmetric(sym):
  function build_rotation (line 78) | def build_rotation(r):
  function build_scaling_rotation (line 101) | def build_scaling_rotation(s, r):
  function safe_state (line 112) | def safe_state(silent):

FILE: utils/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 getProjectionMatrix (line 51) | def getProjectionMatrix(znear, zfar, fovX, fovY):
  function fov2focal (line 73) | def fov2focal(fov, pixels):
  function focal2fov (line 76) | def focal2fov(focal, pixels):

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

FILE: utils/loss_utils.py
  function l1_loss (line 17) | def l1_loss(network_output, gt, mask=None, weight=None):
  function l2_loss (line 25) | def l2_loss(network_output, gt, mask=None, weight=None):
  function gaussian (line 33) | def gaussian(window_size, sigma):
  function create_window (line 37) | def create_window(window_size, channel):
  function ssim (line 43) | def ssim(img1, img2, window_size=11, size_average=True):
  function _ssim (line 53) | def _ssim(img1, img2, window, window_size, channel, size_average=True):

FILE: utils/opengs_utlis.py
  function calculate_pairwise_distances (line 8) | def calculate_pairwise_distances(tensor1, tensor2, metric=None):
  function calculate_distances (line 36) | def calculate_distances(tensor1, tensor2, metric=None):
  function bin2dec (line 61) | def bin2dec(b, bits):
  function load_code_book (line 68) | def load_code_book(base_path):
  function calculate_iou (line 90) | def calculate_iou(masks1, masks2, base=None):
  function get_SAM_mask_and_feat (line 125) | def get_SAM_mask_and_feat(gt_sam_mask, level=3, filter_th=50, original_m...
  function pair_mask_feature_mean (line 184) | def pair_mask_feature_mean(feat_map, masks):
  function process_in_chunks (line 203) | def process_in_chunks(masks_expanded, masked_feats, mean_per_channel, ch...
  function calculate_variance_in_chunks (line 216) | def calculate_variance_in_chunks(masked_for_variance, mask_counts, chunk...
  function ele_multip_in_chunks (line 228) | def ele_multip_in_chunks(feat_expanded, masks_expanded, chunk_size=5):
  function mask_feature_mean (line 240) | def mask_feature_mean(feat_map, gt_masks, image_mask=None, return_var=Fa...
  function linear_to_srgb (line 285) | def linear_to_srgb(linear):
  function srgb_to_linear (line 300) | def srgb_to_linear(srgb):

FILE: utils/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: utils/system_utils.py
  function mkdir_p (line 16) | def mkdir_p(folder_path):
  function searchForMaxIteration (line 26) | def searchForMaxIteration(folder):