SYMBOL INDEX (139 symbols across 13 files) FILE: Spline.py function skew_symmetric (line 7) | def skew_symmetric(w): function taylor_A (line 21) | def taylor_A(x, nth=10): function taylor_B (line 32) | def taylor_B(x, nth=10): function taylor_C (line 42) | def taylor_C(x, nth=10): function exp_r2q_parallel (line 52) | def exp_r2q_parallel(r, eps=1e-9): function exp_r2q (line 61) | def exp_r2q(x, y, z, theta): function exp_r2q_taylor (line 70) | def exp_r2q_taylor(x, y, z, theta): function q_to_R_parallel (line 78) | def q_to_R_parallel(q): function q_to_Q_parallel (line 112) | def q_to_Q_parallel(q): function q_to_q_conj_parallel (line 122) | def q_to_q_conj_parallel(q): function log_q2r_parallel (line 128) | def log_q2r_parallel(q, eps_theta=1e-20, eps_w=1e-10): function log_q2r (line 156) | def log_q2r(w, theta): function log_q2r_taylor_theta_0 (line 160) | def log_q2r_taylor_theta_0(w, theta): function log_q2r_lim_w_0_left (line 164) | def log_q2r_lim_w_0_left(theta): function log_q2r_lim_w_0_right (line 168) | def log_q2r_lim_w_0_right(theta): function SE3_to_se3 (line 172) | def SE3_to_se3(Rt, eps=1e-8): # [...,3,4] function SO3_to_so3 (line 186) | def SO3_to_so3(R, eps=1e-7): # [...,3,3] function se3_to_SE3 (line 199) | def se3_to_SE3(wu): # [...,3] function SE3_to_se3_N (line 213) | def SE3_to_se3_N(poses_rt): function se3_to_SE3_N (line 222) | def se3_to_SE3_N(poses_wu): function se3_2_qt_parallel (line 231) | def se3_2_qt_parallel(wu): function SplineN_linear (line 246) | def SplineN_linear(start_pose, end_pose, poses_number, NUM, device=None): function SplineN_cubic (line 276) | def SplineN_cubic(pose0, pose1, pose2, pose3, poses_number, NUM): FILE: load_llff.py function _minify (line 10) | def _minify(basedir, factors=[], resolutions=[]): # basedir: ./data/ner... function _load_data (line 62) | def _load_data(basedir, pose_state, factor=None, width=None, height=None... function normalize (line 125) | def normalize(x): function viewmatrix (line 129) | def viewmatrix(z, up, pos): function ptstocam (line 138) | def ptstocam(pts, c2w): function poses_avg (line 143) | def poses_avg(poses): function render_path_spiral (line 154) | def render_path_spiral(c2w, up, rads, focal, zdelta, zrate, rots, N): function recenter_poses (line 166) | def recenter_poses(poses): function spherify_poses (line 183) | def spherify_poses(poses, bds): function load_llff_data (line 241) | def load_llff_data(basedir, pose_state=None, factor=1, recenter=True, bd... FILE: lpips/lpips.py function normalize_tensor (line 12) | def normalize_tensor(in_feat, eps=1e-10): function l2 (line 17) | def l2(p0, p1, range=255.): function psnr (line 21) | def psnr(p0, p1, peak=255.): function dssim (line 25) | def dssim(p0, p1, range=255.): function rgb2lab (line 30) | def rgb2lab(in_img, mean_cent=False): function tensor2np (line 38) | def tensor2np(tensor_obj): function np2tensor (line 43) | def np2tensor(np_obj): function tensor2tensorlab (line 48) | def tensor2tensorlab(image_tensor, to_norm=True, mc_only=False): function tensorlab2tensor (line 63) | def tensorlab2tensor(lab_tensor, return_inbnd=False): function tensor2im (line 82) | def tensor2im(image_tensor, imtype=np.uint8, cent=1., factor=255. / 2.): function im2tensor (line 88) | def im2tensor(image, imtype=np.uint8, cent=1., factor=255. / 2.): function tensor2vec (line 93) | def tensor2vec(vector_tensor): function voc_ap (line 97) | def voc_ap(rec, prec, use_07_metric=False): function spatial_average (line 130) | def spatial_average(in_tens, keepdim=True): function upsample (line 134) | def upsample(in_tens, out_HW=(64, 64)): # assumes scale factor is same ... class LPIPS (line 140) | class LPIPS(nn.Module): method __init__ (line 141) | def __init__(self, pretrained=True, net='alex', version='0.1', lpips=T... method forward (line 199) | def forward(self, in0, in1, retPerLayer=False, normalize=False): class ScalingLayer (line 245) | class ScalingLayer(nn.Module): method __init__ (line 246) | def __init__(self): method forward (line 251) | def forward(self, inp): class NetLinLayer (line 255) | class NetLinLayer(nn.Module): method __init__ (line 258) | def __init__(self, chn_in, chn_out=1, use_dropout=False): method forward (line 265) | def forward(self, x): class Dist2LogitLayer (line 269) | class Dist2LogitLayer(nn.Module): method __init__ (line 272) | def __init__(self, chn_mid=32, use_sigmoid=True): method forward (line 284) | def forward(self, d0, d1, eps=0.1): class BCERankingLoss (line 288) | class BCERankingLoss(nn.Module): method __init__ (line 289) | def __init__(self, chn_mid=32): method forward (line 295) | def forward(self, d0, d1, judge): class FakeNet (line 302) | class FakeNet(nn.Module): method __init__ (line 303) | def __init__(self, use_gpu=True, colorspace='Lab'): class L2 (line 309) | class L2(FakeNet): method forward (line 310) | def forward(self, in0, in1, retPerLayer=None): class DSSIM (line 328) | class DSSIM(FakeNet): method forward (line 330) | def forward(self, in0, in1, retPerLayer=None): function print_network (line 346) | def print_network(net): FILE: lpips/pretrained_networks.py class squeezenet (line 6) | class squeezenet(torch.nn.Module): method __init__ (line 7) | def __init__(self, requires_grad=False, pretrained=True): method forward (line 36) | def forward(self, X): class alexnet (line 57) | class alexnet(torch.nn.Module): method __init__ (line 58) | def __init__(self, requires_grad=False, pretrained=True): method forward (line 81) | def forward(self, X): class vgg16 (line 98) | class vgg16(torch.nn.Module): method __init__ (line 99) | def __init__(self, requires_grad=False, pretrained=True): method forward (line 122) | def forward(self, X): class resnet (line 139) | class resnet(torch.nn.Module): method __init__ (line 140) | def __init__(self, requires_grad=False, pretrained=True, num=18): method forward (line 163) | def forward(self, X): FILE: metrics.py function compute_img_metric (line 16) | def compute_img_metric(im1t: torch.Tensor, im2t: torch.Tensor, FILE: nerf.py class Embedder (line 15) | class Embedder: method __init__ (line 16) | def __init__(self, **kwargs): method create_embedding_fn (line 20) | def create_embedding_fn(self): method embed (line 44) | def embed(self, inputs): function get_embedder (line 48) | def get_embedder(args, multires, i=0): class Model (line 66) | class Model(): method __init__ (line 67) | def __init__(self): method build_network (line 70) | def build_network(self, args): method setup_optimizer (line 75) | def setup_optimizer(self, args): class NeRF (line 84) | class NeRF(nn.Module): method __init__ (line 85) | def __init__(self, D=8, W=256, input_ch=63, input_ch_views=27, output_... method forward (line 108) | def forward(self, barf_i, pts, viewdirs, args): method raw2output (line 148) | def raw2output(self, raw, z_vals, rays_d, raw_noise_std=0.0): class Graph (line 176) | class Graph(nn.Module): method __init__ (line 178) | def __init__(self, args, D=8, W=256, input_ch=63, input_ch_views=27, o... method forward (line 184) | def forward(self, i, img_idx, poses_num, H, W, K, args, novel_view=Fal... method get_pose (line 217) | def get_pose(self, i, img_idx, args): method get_gt_pose (line 221) | def get_gt_pose(self, poses, args): method render (line 225) | def render(self, barf_i, poses, ray_idx, H, W, K, args, near=0., far=1... method render_video (line 318) | def render_video(self, barf_i, poses, H, W, K, args): FILE: novel_view_test.py class Model (line 5) | class Model(nerf.Model): method __init__ (line 6) | def __init__(self, se3_start, graph): method build_network (line 11) | def build_network(self, args): method setup_optimizer (line 17) | def setup_optimizer(self, args): FILE: optimize_pose_cubic.py class Model (line 7) | class Model(nerf.Model): method __init__ (line 8) | def __init__(self, se3_0, se3_1, se3_2, se3_3): method build_network (line 15) | def build_network(self, args): method setup_optimizer (line 24) | def setup_optimizer(self, args): class Graph (line 36) | class Graph(nerf.Graph): method __init__ (line 37) | def __init__(self, args, D=8, W=256, input_ch=63, input_ch_views=27, o... method get_pose (line 43) | def get_pose(self, i, img_idx, args): method get_pose_even (line 60) | def get_pose_even(self, i, img_idx, num): method get_gt_pose (line 78) | def get_gt_pose(self, poses, args): FILE: optimize_pose_linear.py class Model (line 7) | class Model(nerf.Model): method __init__ (line 8) | def __init__(self, se3_start, se3_end): method build_network (line 13) | def build_network(self, args): method setup_optimizer (line 22) | def setup_optimizer(self, args): class Graph (line 34) | class Graph(nerf.Graph): method __init__ (line 35) | def __init__(self, args, D=8, W=256, input_ch=63, input_ch_views=27, o... method get_pose (line 41) | def get_pose(self, i, img_idx, args): method get_pose_even (line 53) | def get_pose_even(self, i, img_idx, num): method get_gt_pose (line 66) | def get_gt_pose(self, poses, args): FILE: run_nerf.py function config_parser (line 24) | def config_parser(): FILE: run_nerf_helpers.py function imread (line 19) | def imread(f): function load_imgs (line 25) | def load_imgs(path): function get_rays (line 38) | def get_rays(H, W, K, c2w): function ndc_rays (line 50) | def ndc_rays(H, W, focal, near, rays_o, rays_d): function sample_pdf (line 71) | def sample_pdf(bins, weights, N_samples, det=False, pytest=False): function render_video_test (line 117) | def render_video_test(i_, graph, render_poses, H, W, K, args): function render_image_test (line 139) | def render_image_test(i, graph, render_poses, H, W, K, args, novel_view=... function init_weights (line 165) | def init_weights(linear): function init_nerf (line 171) | def init_nerf(nerf): function get_specific_rays (line 185) | def get_specific_rays(i, j, K, c2w): function save_render_pose (line 199) | def save_render_pose(poses, path): FILE: test.py function test (line 13) | def test(): FILE: train.py function train (line 13) | def train():