SYMBOL INDEX (210 symbols across 24 files)

FILE: distill_mutual/network.py
  class NeRFNetwork (line 12) | class NeRFNetwork(NeRFRenderer):
    method __init__ (line 13) | def __init__(
    method init_plenoxel_volume (line 184) | def init_plenoxel_volume(self, s=0.1, fea_dim=27 + 1, volume=[128, 128...
    method init_one_vm (line 193) | def init_one_vm(self, n_component, resolution, scale=0.1):
    method get_sigma_feat (line 216) | def get_sigma_feat(self, x):
    method get_color_feat (line 264) | def get_color_feat(self, x):
    method compute_plenoxel_fea (line 311) | def compute_plenoxel_fea(self, x):
    method forward_nerf_mlp (line 324) | def forward_nerf_mlp(self, x):
    method forward (line 335) | def forward(self, x, d):
    method density (line 439) | def density(self, x):
    method background (line 496) | def background(self, x, d):
    method color (line 515) | def color(self, x, d, mask=None, geo_feat=None, **kwargs):
    method density_loss (line 549) | def density_loss(self):
    method upsample_params (line 561) | def upsample_params(self, mat, vec, resolution):
    method upsample_model (line 584) | def upsample_model(self, resolution):
    method shrink_model (line 590) | def shrink_model(self):
    method get_params (line 646) | def get_params(self, lr, lr2=1e-3):

FILE: distill_mutual/provider.py
  function nerf_matrix_to_ngp (line 18) | def nerf_matrix_to_ngp(pose, scale=0.33):
  function rand_poses (line 32) | def rand_poses(
  class NeRFDataset (line 123) | class NeRFDataset:
    method __init__ (line 124) | def __init__(self, opt, device, type="train", downscale=1, n_test=10):
    method collate (line 284) | def collate(self, index):
    method dataloader (line 316) | def dataloader(self):

FILE: distill_mutual/renderer.py
  function sample_pdf (line 15) | def sample_pdf(bins, weights, n_samples, det=False):
  function plot_pointcloud (line 54) | def plot_pointcloud(pc, color=None):
  class NeRFRenderer (line 66) | class NeRFRenderer(nn.Module):
    method __init__ (line 67) | def __init__(
    method forward (line 117) | def forward(self, x, d):
    method density (line 121) | def density(self, x):
    method color (line 124) | def color(self, x, d, mask=None, **kwargs):
    method reset_extra_state (line 127) | def reset_extra_state(self):
    method run (line 139) | def run(
    method run_cuda (line 319) | def run_cuda(
    method mark_untrained_grid (line 562) | def mark_untrained_grid(self, poses, intrinsic, S=64):
    method update_extra_state (line 648) | def update_extra_state(self, decay=0.95, S=128):
    method render (line 777) | def render(self, rays_o, rays_d, staged=False, max_ray_batch=4096, **k...

FILE: distill_mutual/utils.py
  function update_loss_rate (line 41) | def update_loss_rate(cur_lrate, scale=0.99):
  function get_softmax_map_mean (line 45) | def get_softmax_map_mean(a, b):
  function get_kl (line 49) | def get_kl(inputs, targets):
  function nerf_matrix_to_ngp (line 53) | def nerf_matrix_to_ngp(pose, scale=0.8):
  function pose_spherical (line 67) | def pose_spherical(theta, phi, radius):
  function get_rand_poses (line 100) | def get_rand_poses(data_type="synthetic", original_loader=None):
  function custom_meshgrid (line 201) | def custom_meshgrid(*args):
  function linear_to_srgb (line 210) | def linear_to_srgb(x):
  function srgb_to_linear (line 215) | def srgb_to_linear(x):
  function compute_ssim (line 219) | def compute_ssim(
  function init_lpips (line 303) | def init_lpips(net_name, device):
  function rgb_lpips (line 317) | def rgb_lpips(gt, im, net_name):
  function get_rays (line 325) | def get_rays(poses, intrinsics, H, W, N=-1, error_map=None):
  function seed_everything (line 407) | def seed_everything(seed):
  function torch_vis_2d (line 417) | def torch_vis_2d(x, renormalize=False):
  function extract_fields (line 442) | def extract_fields(bound_min, bound_max, resolution, query_func, S=128):
  function extract_geometry (line 473) | def extract_geometry(bound_min, bound_max, resolution, threshold, query_...
  class PSNRMeter (line 491) | class PSNRMeter:
    method __init__ (line 492) | def __init__(self):
    method clear (line 497) | def clear(self):
    method prepare_inputs (line 502) | def prepare_inputs(self, *inputs):
    method update (line 511) | def update(self, preds, truths):
    method measure (line 522) | def measure(self):
    method write (line 525) | def write(self, writer, global_step, prefix=""):
    method report (line 528) | def report(self):
  class Trainer (line 532) | class Trainer(object):
    method __init__ (line 533) | def __init__(
    method __del__ (line 672) | def __del__(self):
    method log (line 676) | def log(self, *args, **kwargs):
    method train (line 685) | def train(self, train_loader, valid_loader, max_epochs):
    method train_one_epoch (line 753) | def train_one_epoch(self, loader):
    method get_loss (line 941) | def get_loss(self, pred, gt):
    method train_step (line 954) | def train_step(self, data):
    method evaluate (line 1193) | def evaluate(self, loader, name=None):
    method evaluate_one_epoch (line 1198) | def evaluate_one_epoch(self, loader, name=None):
    method eval_step (line 1370) | def eval_step(self, data):
    method save_checkpoint (line 1405) | def save_checkpoint(self, name=None, full=False, best=False, remove_ol...
    method load_teacher_checkpoint (line 1477) | def load_teacher_checkpoint(self):
    method load_student_checkpoint (line 1531) | def load_student_checkpoint(self):
    method load_checkpoint (line 1589) | def load_checkpoint(self, checkpoint=None, model_only=False):
    method test (line 1653) | def test(self, loader, save_path=None, name=None):
    method test_step (line 1703) | def test_step(self, data, bg_color=None, perturb=False):

FILE: gridencoder/backend.py
  function find_cl_path (line 20) | def find_cl_path():

FILE: gridencoder/grid.py
  class _grid_encode (line 20) | class _grid_encode(Function):
    method forward (line 23) | def forward(
    method backward (line 96) | def backward(ctx, grad):
  class GridEncoder (line 142) | class GridEncoder(nn.Module):
    method __init__ (line 143) | def __init__(
    method reset_parameters (line 200) | def reset_parameters(self):
    method __repr__ (line 204) | def __repr__(self):
    method forward (line 207) | def forward(self, inputs, bound=1):

FILE: gridencoder/setup.py
  function find_cl_path (line 21) | def find_cl_path():

FILE: gridencoder/src/bindings.cpp
  function PYBIND11_MODULE (line 5) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: just_train_tea/network.py
  class NeRFNetwork (line 12) | class NeRFNetwork(NeRFRenderer):
    method __init__ (line 13) | def __init__(
    method init_plenoxel_volume (line 184) | def init_plenoxel_volume(self, s=0.1, fea_dim=27 + 1, volume=[128, 128...
    method init_one_vm (line 193) | def init_one_vm(self, n_component, resolution, scale=0.1):
    method get_sigma_feat (line 216) | def get_sigma_feat(self, x):
    method get_color_feat (line 264) | def get_color_feat(self, x):
    method compute_plenoxel_fea (line 311) | def compute_plenoxel_fea(self, x):
    method forward_nerf_mlp (line 320) | def forward_nerf_mlp(self, x):
    method forward (line 331) | def forward(self, x, d):
    method density (line 422) | def density(self, x):
    method background (line 479) | def background(self, x, d):
    method color (line 498) | def color(self, x, d, mask=None, geo_feat=None, **kwargs):
    method density_loss (line 532) | def density_loss(self):
    method upsample_params (line 544) | def upsample_params(self, mat, vec, resolution):
    method upsample_model (line 567) | def upsample_model(self, resolution):
    method shrink_model (line 573) | def shrink_model(self):
    method get_params (line 628) | def get_params(self, lr, lr2=1e-3):

FILE: just_train_tea/provider.py
  function nerf_matrix_to_ngp (line 18) | def nerf_matrix_to_ngp(pose, scale=0.33):
  function rand_poses (line 32) | def rand_poses(
  class NeRFDataset (line 123) | class NeRFDataset:
    method __init__ (line 124) | def __init__(self, opt, device, type="train", downscale=1, n_test=10):
    method collate (line 284) | def collate(self, index):
    method dataloader (line 316) | def dataloader(self):

FILE: just_train_tea/renderer.py
  function sample_pdf (line 14) | def sample_pdf(bins, weights, n_samples, det=False):
  function plot_pointcloud (line 53) | def plot_pointcloud(pc, color=None):
  class NeRFRenderer (line 65) | class NeRFRenderer(nn.Module):
    method __init__ (line 66) | def __init__(
    method forward (line 116) | def forward(self, x, d):
    method density (line 120) | def density(self, x):
    method color (line 123) | def color(self, x, d, mask=None, **kwargs):
    method reset_extra_state (line 126) | def reset_extra_state(self):
    method run (line 138) | def run(
    method run_cuda (line 319) | def run_cuda(
    method mark_untrained_grid (line 555) | def mark_untrained_grid(self, poses, intrinsic, S=64):
    method update_extra_state (line 641) | def update_extra_state(self, decay=0.95, S=128):
    method render (line 770) | def render(self, rays_o, rays_d, staged=False, max_ray_batch=4096, **k...

FILE: just_train_tea/utils.py
  function custom_meshgrid (line 36) | def custom_meshgrid(*args):
  function linear_to_srgb (line 45) | def linear_to_srgb(x):
  function srgb_to_linear (line 50) | def srgb_to_linear(x):
  function compute_ssim (line 54) | def compute_ssim(
  function init_lpips (line 138) | def init_lpips(net_name, device):
  function rgb_lpips (line 152) | def rgb_lpips(gt, im, net_name):
  function get_rays (line 160) | def get_rays(poses, intrinsics, H, W, N=-1, error_map=None):
  function seed_everything (line 242) | def seed_everything(seed):
  function torch_vis_2d (line 252) | def torch_vis_2d(x, renormalize=False):
  function extract_fields (line 277) | def extract_fields(bound_min, bound_max, resolution, query_func, S=128):
  function extract_geometry (line 308) | def extract_geometry(bound_min, bound_max, resolution, threshold, query_...
  class PSNRMeter (line 326) | class PSNRMeter:
    method __init__ (line 327) | def __init__(self):
    method clear (line 331) | def clear(self):
    method prepare_inputs (line 335) | def prepare_inputs(self, *inputs):
    method update (line 344) | def update(self, preds, truths):
    method measure (line 355) | def measure(self):
    method write (line 358) | def write(self, writer, global_step, prefix=""):
    method report (line 361) | def report(self):
  class Trainer (line 365) | class Trainer(object):
    method __init__ (line 366) | def __init__(
    method __del__ (line 487) | def __del__(self):
    method log (line 491) | def log(self, *args, **kwargs):
    method train (line 500) | def train(self, train_loader, valid_loader, max_epochs):
    method train_one_epoch (line 543) | def train_one_epoch(self, loader):
    method get_loss (line 733) | def get_loss(self, pred, gt):
    method train_step (line 746) | def train_step(self, data):
    method evaluate (line 848) | def evaluate(self, loader, name=None):
    method evaluate_one_epoch (line 853) | def evaluate_one_epoch(self, loader, name=None):
    method eval_step (line 1028) | def eval_step(self, data):
    method save_checkpoint (line 1063) | def save_checkpoint(self, name=None, full=False, best=False, remove_ol...
    method load_teacher_checkpoint (line 1135) | def load_teacher_checkpoint(self):
    method load_student_checkpoint (line 1158) | def load_student_checkpoint(self):
    method test (line 1187) | def test(self, loader, save_path=None, name=None):
    method test_step (line 1237) | def test_step(self, data, bg_color=None, perturb=False):

FILE: main_distill_mutual.py
  function save_codes_env (line 15) | def save_codes_env(workspace):
  function load_from_txt (line 24) | def load_from_txt(opt, except_space=""):

FILE: raymarching/backend.py
  function find_cl_path (line 20) | def find_cl_path():

FILE: raymarching/raymarching.py
  class _near_far_from_aabb (line 20) | class _near_far_from_aabb(Function):
    method forward (line 23) | def forward(ctx, rays_o, rays_d, aabb, min_near=0.2):
  class _polar_from_ray (line 56) | class _polar_from_ray(Function):
    method forward (line 59) | def forward(ctx, rays_o, rays_d, radius):
  class _morton3D (line 90) | class _morton3D(Function):
    method forward (line 92) | def forward(ctx, coords):
  class _morton3D_invert (line 116) | class _morton3D_invert(Function):
    method forward (line 118) | def forward(ctx, indices):
  class _packbits (line 141) | class _packbits(Function):
    method forward (line 144) | def forward(ctx, grid, thresh, bitfield=None):
  class _march_rays_train (line 176) | class _march_rays_train(Function):
    method forward (line 179) | def forward(
  class _composite_rays_train (line 292) | class _composite_rays_train(Function):
    method forward (line 295) | def forward(ctx, sigmas, rgbs, deltas, rays):
    method backward (line 329) | def backward(ctx, grad_weights_sum, grad_depth, grad_image):
  class _march_rays (line 367) | class _march_rays(Function):
    method forward (line 370) | def forward(
  class _composite_rays (line 457) | class _composite_rays(Function):
    method forward (line 460) | def forward(
  class _compact_rays (line 505) | class _compact_rays(Function):
    method forward (line 508) | def forward(

FILE: raymarching/setup.py
  function find_cl_path (line 21) | def find_cl_path():

FILE: raymarching/src/bindings.cpp
  function PYBIND11_MODULE (line 5) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: raymarching/src/pcg32.h
  type pcg32 (line 44) | struct pcg32 {
  function next_uint (line 66) | uint32_t next_uint() {
  function next_uint (line 75) | uint32_t next_uint(uint32_t bound) {
  function next_float (line 107) | float next_float() {
  function next_double (line 125) | double next_double() {
  function operator (line 198) | bool operator==(const pcg32 &other) const { return state == other.state ...
  function operator (line 201) | bool operator!=(const pcg32 &other) const { return state != other.state ...

FILE: shencoder/backend.py
  function find_cl_path (line 20) | def find_cl_path():

FILE: shencoder/setup.py
  function find_cl_path (line 21) | def find_cl_path():

FILE: shencoder/sphere_harmonics.py
  class _sh_encoder (line 15) | class _sh_encoder(Function):
    method forward (line 18) | def forward(ctx, inputs, degree, calc_grad_inputs=False):
    method backward (line 48) | def backward(ctx, grad):
  class SHEncoder (line 67) | class SHEncoder(nn.Module):
    method __init__ (line 68) | def __init__(self, input_dim=3, degree=4):
    method __repr__ (line 80) | def __repr__(self):
    method forward (line 83) | def forward(self, inputs, size=1):

FILE: shencoder/src/bindings.cpp
  function PYBIND11_MODULE (line 5) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: tools/activation.py
  class _trunc_exp (line 6) | class _trunc_exp(Function):
    method forward (line 9) | def forward(ctx, x):
    method backward (line 15) | def backward(ctx, g):

FILE: tools/encoding.py
  class FreqEncoder (line 6) | class FreqEncoder(nn.Module):
    method __init__ (line 7) | def __init__(
    method forward (line 36) | def forward(self, input, **kwargs):
  function get_encoder (line 52) | def get_encoder(