SYMBOL INDEX (246 symbols across 21 files)

FILE: dataLoader/blender.py
  class BlenderDataset (line 13) | class BlenderDataset(Dataset):
    method __init__ (line 14) | def __init__(self, datadir, split='train', downsample=1.0, is_stack=Fa...
    method read_depth (line 35) | def read_depth(self, filename):
    method read_meta (line 39) | def read_meta(self):
    method prepare_feature_data (line 103) | def prepare_feature_data(self, encoder, chunk=8):
    method define_transforms (line 125) | def define_transforms(self):
    method define_proj_mat (line 128) | def define_proj_mat(self):
    method world2ndc (line 131) | def world2ndc(self,points,lindisp=None):
    method __len__ (line 135) | def __len__(self):
    method __getitem__ (line 138) | def __getitem__(self, idx):

FILE: dataLoader/colmap2nerf.py
  function parse_args (line 23) | def parse_args():
  function do_system (line 40) | def do_system(arg):
  function run_ffmpeg (line 47) | def run_ffmpeg(args):
  function run_colmap (line 69) | def run_colmap(args):
  function variance_of_laplacian (line 99) | def variance_of_laplacian(image):
  function sharpness (line 102) | def sharpness(imagePath):
  function qvec2rotmat (line 108) | def qvec2rotmat(qvec):
  function rotmat (line 125) | def rotmat(a, b):
  function closest_point_2_lines (line 133) | def closest_point_2_lines(oa, da, ob, db): # returns point closest to bo...

FILE: dataLoader/llff.py
  function normalize (line 12) | def normalize(v):
  function average_poses (line 17) | def average_poses(poses):
  function center_poses (line 54) | def center_poses(poses, blender2opencv):
  function viewmatrix (line 81) | def viewmatrix(z, up, pos):
  function render_path_spiral (line 91) | def render_path_spiral(c2w, up, rads, focal, zdelta, zrate, N_rots=2, N=...
  function get_spiral (line 102) | def get_spiral(c2ws_all, near_fars, rads_scale=1.0, N_views=120):
  function get_interpolation_path (line 121) | def get_interpolation_path(c2ws_all, steps=30):
  class LLFFDataset (line 147) | class LLFFDataset(Dataset):
    method __init__ (line 148) | def __init__(self, datadir, split='train', downsample=4, is_stack=Fals...
    method read_meta (line 168) | def read_meta(self):
    method prepare_feature_data (line 257) | def prepare_feature_data(self, encoder, chunk=8):
    method define_transforms (line 279) | def define_transforms(self):
    method __len__ (line 282) | def __len__(self):
    method __getitem__ (line 285) | def __getitem__(self, idx):

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

FILE: dataLoader/ray_utils.py
  function depth2dist (line 9) | def depth2dist(z_vals, cos_angle):
  function ndc2dist (line 18) | def ndc2dist(ndc_pts, cos_angle):
  function get_ray_directions (line 24) | def get_ray_directions(H, W, focal, center=None):
  function get_ray_directions_blender (line 45) | def get_ray_directions_blender(H, W, focal, center=None):
  function get_rays (line 66) | def get_rays(directions, c2w):
  function ndc_rays_blender (line 90) | def ndc_rays_blender(H, W, focal, near, rays_o, rays_d):
  function ndc_rays (line 109) | def ndc_rays(H, W, focal, near, rays_o, rays_d):
  function sample_pdf (line 129) | def sample_pdf(bins, weights, N_samples, det=False, pytest=False):
  function dda (line 174) | def dda(rays_o, rays_d, bbox_3D):
  function ray_marcher (line 184) | def ray_marcher(rays,
  function read_pfm (line 231) | def read_pfm(filename):
  function ndc_bbox (line 269) | def ndc_bbox(all_rays):
  function denormalize_vgg (line 281) | def denormalize_vgg(img):

FILE: dataLoader/styleLoader.py
  function getDataLoader (line 6) | def getDataLoader(dataset_path, batch_size, sampler, image_side_length=2...

FILE: dataLoader/tankstemple.py
  function circle (line 12) | def circle(radius=3.5, h=0.0, axis='z', t0=0, r=1):
  function cross (line 21) | def cross(x, y, axis=0):
  function normalize (line 26) | def normalize(x, axis=-1, order=2):
  function cat (line 38) | def cat(x, axis=1):
  function look_at_rotation (line 44) | def look_at_rotation(camera_position, at=None, up=None, inverse=False, c...
  function gen_path (line 77) | def gen_path(pos_gen, at=(0, 0, 0), up=(0, -1, 0), frames=180):
  class TanksTempleDataset (line 87) | class TanksTempleDataset(Dataset):
    method __init__ (line 89) | def __init__(self, datadir, split='train', downsample=4.0, wh=[1920,10...
    method bbox2corners (line 108) | def bbox2corners(self):
    method read_meta (line 115) | def read_meta(self):
    method prepare_feature_data (line 196) | def prepare_feature_data(self, encoder, chunk=4):
    method define_transforms (line 219) | def define_transforms(self):
    method define_proj_mat (line 222) | def define_proj_mat(self):
    method world2ndc (line 225) | def world2ndc(self, points):
    method __len__ (line 229) | def __len__(self):
    method __getitem__ (line 234) | def __getitem__(self, idx):

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

FILE: extra/auto_run_paramsets.py
  function getFolderLocker (line 7) | def getFolderLocker(logFolder):
  function releaseFolderLocker (line 15) | def releaseFolderLocker(logFolder):
  function getStopFolder (line 18) | def getStopFolder(logFolder):
  function get_param_str (line 22) | def get_param_str(key, val):
  function get_param_list (line 28) | def get_param_list(param_dict):
  function run_program (line 167) | def run_program(gpu, expname, param):

FILE: extra/compute_metrics.py
  function init_lpips (line 11) | def init_lpips(net_name, device):
  function rgb_lpips (line 17) | def rgb_lpips(np_gt, np_im, net_name, device):
  function findItem (line 25) | def findItem(items, target):
  function rgb_ssim (line 34) | def rgb_ssim(img0, img1, max_val,

FILE: models/VGG.py
  class Encoder (line 8) | class Encoder(nn.Module):
    method __init__ (line 9) | def __init__(self):
    method forward (line 23) | def forward(self, x):
  class Decoder (line 34) | class Decoder(nn.Module):
    method __init__ (line 38) | def __init__(self, ckpt_path=None):
    method forward (line 67) | def forward(self, x):
  class DownBlock (line 74) | class DownBlock(nn.Module):
    method __init__ (line 76) | def __init__(self, in_dim, out_dim, down='conv'):
    method forward (line 93) | def forward(self, x):
  class UpBlock (line 98) | class UpBlock(nn.Module):
    method __init__ (line 100) | def __init__(self, in_dim, out_dim, skip_dim=None, up='nearest'):
    method _pad (line 125) | def _pad(self, x, y):
    method forward (line 142) | def forward(self, x, skip=None):
  class UNetDecoder (line 150) | class UNetDecoder(nn.Module):
    method __init__ (line 152) | def __init__(self, in_dim=256):
    method forward (line 184) | def forward(self, feats):
  class PlainDecoder (line 197) | class PlainDecoder(nn.Module):
    method __init__ (line 198) | def __init__(self) -> None:
    method forward (line 219) | def forward(self, x):

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

FILE: models/styleModules.py
  function calc_mean_std (line 6) | def calc_mean_std(x, eps=1e-8):
  function cal_adain_style_loss (line 16) | def cal_adain_style_loss(x, y):
  function cal_mse_content_loss (line 29) | def cal_mse_content_loss(x, y):
  class LearnableIN (line 34) | class LearnableIN(nn.Module):
    method __init__ (line 38) | def __init__(self, dim=256):
    method forward (line 42) | def forward(self, x):
  class SimpleLinearStylizer (line 47) | class SimpleLinearStylizer(nn.Module):
    method __init__ (line 48) | def __init__(self, input_dim=256, embed_dim=32, n_layers=3 ) -> None:
    method _vectorized_covariance (line 76) | def _vectorized_covariance(self, x):
    method get_content_matrix (line 81) | def get_content_matrix(self, c):
    method get_style_mean_std_matrix (line 98) | def get_style_mean_std_matrix(self, s):
    method transform_content_3D (line 117) | def transform_content_3D(self, c):
    method transfer_style_2D (line 131) | def transfer_style_2D(self, s_mean_std_mat, c, acc_map):
  class AdaAttN (line 151) | class AdaAttN(nn.Module):
    method __init__ (line 154) | def __init__(self, qk_dim, v_dim):
    method forward (line 166) | def forward(self, q, k):
  class AdaAttN_new_IN (line 199) | class AdaAttN_new_IN(nn.Module):
    method __init__ (line 202) | def __init__(self, qk_dim, v_dim):
    method forward (line 215) | def forward(self, q, k):
  class AdaAttN_woin (line 248) | class AdaAttN_woin(nn.Module):
    method __init__ (line 251) | def __init__(self, qk_dim, v_dim):
    method forward (line 263) | def forward(self, q, k):

FILE: models/tensoRF.py
  class TensorVMSplit (line 5) | class TensorVMSplit(TensorBase):
    method __init__ (line 6) | def __init__(self, aabb, gridSize, device, **kargs):
    method change_to_feature_mod (line 11) | def change_to_feature_mod(self, feature_n_comp, device):
    method change_to_style_mod (line 29) | def change_to_style_mod(self, device='cuda'):
    method init_svd_volume (line 45) | def init_svd_volume(self, res, device):
    method init_feature_svd (line 50) | def init_feature_svd(self, device):
    method init_one_svd (line 54) | def init_one_svd(self, n_component, gridSize, scale, device):
    method get_optparam_groups (line 68) | def get_optparam_groups(self, lr_init_spatialxyz = 0.02, lr_init_netwo...
    method get_optparam_groups_feature_mod (line 76) | def get_optparam_groups_feature_mod(self, lr_init_spatialxyz, lr_init_...
    method get_optparam_groups_style_mod (line 83) | def get_optparam_groups_style_mod(self, lr_init_network, lr_finetune):
    method vectorDiffs (line 92) | def vectorDiffs(self, vector_comps):
    method vector_comp_diffs (line 103) | def vector_comp_diffs(self):
    method density_L1 (line 106) | def density_L1(self):
    method TV_loss_density (line 112) | def TV_loss_density(self, reg):
    method TV_loss_app (line 118) | def TV_loss_app(self, reg):
    method TV_loss_feature (line 124) | def TV_loss_feature(self, reg):
    method compute_densityfeature (line 132) | def compute_densityfeature(self, xyz_sampled):
    method compute_appfeature (line 149) | def compute_appfeature(self, xyz_sampled):
    method compute_feature (line 167) | def compute_feature(self, xyz_sampled):
    method render_feature_map (line 187) | def render_feature_map(self, rays_chunk, s_mean_std_mat=None, is_train...
    method render_depth_map (line 247) | def render_depth_map(self, rays_chunk, is_train=False, ndc_ray=False, ...
    method up_sampling_VM (line 287) | def up_sampling_VM(self, plane_coef, line_coef, res_target):
    method upsample_volume_grid (line 302) | def upsample_volume_grid(self, res_target):
    method shrink (line 310) | def shrink(self, new_aabb):

FILE: models/tensorBase.py
  function positional_encoding (line 9) | def positional_encoding(positions, freqs):
  function raw2alpha (line 17) | def raw2alpha(sigma, dist):
  function SHRender (line 27) | def SHRender(xyz_sampled, viewdirs, features):
  function RGBRender (line 34) | def RGBRender(xyz_sampled, viewdirs, features):
  class AlphaGridMask (line 39) | class AlphaGridMask(torch.nn.Module):
    method __init__ (line 40) | def __init__(self, device, aabb, alpha_volume):
    method sample_alpha (line 50) | def sample_alpha(self, xyz_sampled):
    method normalize_coord (line 56) | def normalize_coord(self, xyz_sampled):
  class MLPRender_Fea (line 60) | class MLPRender_Fea(torch.nn.Module):
    method __init__ (line 61) | def __init__(self,inChanel, viewpe=6, feape=6, featureC=128):
    method forward (line 74) | def forward(self, pts, viewdirs, features):
  class MLPRender_PE (line 86) | class MLPRender_PE(torch.nn.Module):
    method __init__ (line 87) | def __init__(self,inChanel, viewpe=6, pospe=6, featureC=128):
    method forward (line 100) | def forward(self, pts, viewdirs, features):
  class MLPRender (line 112) | class MLPRender(torch.nn.Module):
    method __init__ (line 113) | def __init__(self,inChanel, viewpe=6, featureC=128):
    method forward (line 126) | def forward(self, pts, viewdirs, features):
  class TensorBase (line 138) | class TensorBase(torch.nn.Module):
    method __init__ (line 139) | def __init__(self, aabb, gridSize, device, density_n_comp = 8, appeara...
    method init_render_func (line 175) | def init_render_func(self, shadingMode, pos_pe, view_pe, fea_pe, featu...
    method update_stepSize (line 192) | def update_stepSize(self, gridSize):
    method init_svd_volume (line 205) | def init_svd_volume(self, res, device):
    method compute_features (line 208) | def compute_features(self, xyz_sampled):
    method compute_densityfeature (line 211) | def compute_densityfeature(self, xyz_sampled):
    method compute_appfeature (line 214) | def compute_appfeature(self, xyz_sampled):
    method normalize_coord (line 217) | def normalize_coord(self, xyz_sampled):
    method get_optparam_groups (line 220) | def get_optparam_groups(self, lr_init_spatial = 0.02, lr_init_network ...
    method get_kwargs (line 223) | def get_kwargs(self):
    method save (line 247) | def save(self, path):
    method load (line 257) | def load(self, ckpt):
    method sample_ray_ndc (line 265) | def sample_ray_ndc(self, rays_o, rays_d, is_train=True, N_samples=-1):
    method sample_ray (line 276) | def sample_ray(self, rays_o, rays_d, is_train=True, N_samples=-1):
    method shrink (line 298) | def shrink(self, new_aabb, voxel_size):
    method getDenseAlpha (line 302) | def getDenseAlpha(self,gridSize=None):
    method updateAlphaMask (line 320) | def updateAlphaMask(self, gridSize=(200,200,200)):
    method filtering_rays (line 346) | def filtering_rays(self, all_rays, all_rgbs, N_samples=256, chunk=1024...
    method feature2density (line 378) | def feature2density(self, density_features):
    method compute_alpha (line 385) | def compute_alpha(self, xyz_locs, length=1):
    method forward (line 408) | def forward(self, rays_chunk, white_bg=True, is_train=False, ndc_ray=F...

FILE: opt.py
  function config_parser (line 3) | def config_parser(cmd=None):

FILE: renderer.py
  function OctreeRender_trilinear_fast (line 9) | def OctreeRender_trilinear_fast(rays, tensorf, chunk=4096, N_samples=-1,...
  function OctreeRender_trilinear_fast_depth (line 40) | def OctreeRender_trilinear_fast_depth(rays, tensorf, chunk=4096, N_sampl...
  function evaluation_feature (line 53) | def evaluation_feature(test_dataset, tensorf, args, renderer, chunk_size...
  function evaluation_feature_path (line 139) | def evaluation_feature_path(test_dataset, tensorf, c2ws, renderer, chunk...
  function evaluation (line 203) | def evaluation(test_dataset,tensorf, args, renderer, savePath=None, N_vi...
  function evaluation_path (line 269) | def evaluation_path(test_dataset,tensorf, c2ws, renderer, savePath=None,...

FILE: train.py
  class SimpleSampler (line 24) | class SimpleSampler:
    method __init__ (line 25) | def __init__(self, total, batch):
    method nextids (line 31) | def nextids(self):
  function export_mesh (line 40) | def export_mesh(args):
  function render_test (line 53) | def render_test(args):
  function reconstruction (line 89) | def reconstruction(args):

FILE: train_feature.py
  class SimpleSampler (line 27) | class SimpleSampler:
    method __init__ (line 28) | def __init__(self, total, batch):
    method nextids (line 34) | def nextids(self):
  function InfiniteSampler (line 41) | def InfiniteSampler(n):
  class InfiniteSamplerWrapper (line 53) | class InfiniteSamplerWrapper(torch.utils.data.sampler.Sampler):
    method __init__ (line 54) | def __init__(self, num_samples):
    method __iter__ (line 57) | def __iter__(self):
    method __len__ (line 60) | def __len__(self):
  function render_test (line 64) | def render_test(args):
  function reconstruction (line 104) | def reconstruction(args):

FILE: train_style.py
  class SimpleSampler (line 30) | class SimpleSampler:
    method __init__ (line 31) | def __init__(self, total, batch):
    method nextids (line 37) | def nextids(self):
  function InfiniteSampler (line 44) | def InfiniteSampler(n):
  class InfiniteSamplerWrapper (line 56) | class InfiniteSamplerWrapper(torch.utils.data.sampler.Sampler):
    method __init__ (line 57) | def __init__(self, num_samples):
    method __iter__ (line 60) | def __iter__(self):
    method __len__ (line 63) | def __len__(self):
  function render_test (line 67) | def render_test(args):
  function reconstruction (line 113) | def reconstruction(args):

FILE: utils.py
  function visualize_depth_numpy (line 11) | def visualize_depth_numpy(depth, minmax=None, cmap=cv2.COLORMAP_JET):
  function init_log (line 28) | def init_log(log, keys):
  function visualize_depth (line 33) | def visualize_depth(depth, minmax=None, cmap=cv2.COLORMAP_JET):
  function N_to_reso (line 53) | def N_to_reso(n_voxels, bbox):
  function cal_n_samples (line 59) | def cal_n_samples(reso, step_ratio=0.5):
  function init_lpips (line 66) | def init_lpips(net_name, device):
  function rgb_lpips (line 72) | def rgb_lpips(np_gt, np_im, net_name, device):
  function findItem (line 80) | def findItem(items, target):
  function rgb_ssim (line 89) | def rgb_ssim(img0, img1, max_val,
  class TVLoss (line 139) | class TVLoss(nn.Module):
    method __init__ (line 140) | def __init__(self):
    method forward (line 143) | def forward(self,x):
    method _tensor_size (line 164) | def _tensor_size(self,t):
  function convert_sdf_samples_to_ply (line 171) | def convert_sdf_samples_to_ply(
  function get_checkerboard (line 355) | def get_checkerboard(fg, n=8):