MCPcopy Create free account

hub / github.com/VITA-Group/INS / functions

Functions223 in github.com/VITA-Group/INS

⨍ Functions 223◇ Types & classes 36
↓ 12 callersFunctionnormalize
(x)
data/load_llff.py:129
↓ 9 callersFunctionto8b
(x)
utils/image.py:40
↓ 7 callersFunctionimg2mse
(x, y, mask=None)
utils/image.py:28
↓ 6 callersFunctionpytorch_phase_shift
(Fx, shift)
utils/pytorch_projector.py:47
↓ 5 callersMethodstep
(self, step)
engines/lr.py:20
↓ 4 callersFunctioninterpolate
(batch, mode='bilinear', size=[224, 224])
idr/model/loss.py:7
↓ 4 callersFunctioninterpolate
(batch, mode='bilinear', size=[224, 224])
engines/trainer.py:22
↓ 4 callersMethodnear_far
(self)
data/datasets.py:86
↓ 4 callersFunctionpytorch_affine_transform
(vol, theta, interp_mode='bilinear')
utils/pytorch_projector.py:58
↓ 4 callersFunctionto_numpy_complex
(x)
utils/pytorch_projector.py:24
↓ 3 callersMethod__init__
(self, root_dir, split='train', subsample=0, cam_id=False, rgb=True, with_mask=False)
data/datasets.py:12
↓ 3 callersMethod__init__
(self, in_channels, out_channels, kernel_size, stride)
models/transformer_net.py:48
↓ 3 callersFunction_minify
(basedir, factors=[], resolutions=[])
data/load_llff.py:8
↓ 3 callersMethodchange_sampling_idx
(self, sampling_size, stride=4)
idr/datasets/scene_dataset.py:122
↓ 3 callersFunctiondir2poses
(posedir)
data/load_deepvoxels.py:65
↓ 3 callersFunctioneval_one_view
Model inference
engines/eval.py:21
↓ 3 callersFunctionevaluate
Main function of evaluation
engines/eval.py:103
↓ 3 callersFunctionimg2mae
(x, y, mask=None)
utils/image.py:34
↓ 3 callersFunctionnormalize_batch
(batch)
idr/model/loss.py:11
↓ 3 callersFunctionnormalize_batch
(batch)
engines/trainer.py:26
↓ 3 callersFunctionposes_avg
(poses)
data/load_llff.py:144
↓ 3 callersFunctionquat_to_rot
(q)
idr/utils/rend_util.py:102
↓ 3 callersFunctionto_pytorch_complex
(x)
utils/pytorch_projector.py:21
↓ 2 callersMethod__init__
(self, conf)
idr/model/implicit_differentiable_renderer.py:160
↓ 2 callersMethod__init__
Init layered sampling init_planes: [N_planes, 4], Ax + By + Cz = D trainable: Whether planes can be trained by optimizer
models/sampler.py:171
↓ 2 callersMethod__init__
MLP backbone for NeRF
models/nerf_mlp.py:19
↓ 2 callersFunctionget_embedder
(multires)
idr/model/embedder.py:38
↓ 2 callersFunctionget_grid_uniform
(resolution)
idr/utils/plots.py:229
↓ 2 callersMethodget_gt_pose
(self, scaled=False)
idr/datasets/scene_dataset.py:136
↓ 2 callersFunctionget_normalization
(source_dir, use_linear_init=False)
idr/preprocess/preprocess_cameras.py:182
↓ 2 callersFunctionget_persp_rays
(H, W, K, c2w, z_dir=-1.)
utils/ray.py:12
↓ 2 callersMethodget_pose_init
(self)
idr/datasets/scene_dataset.py:153
↓ 2 callersMethodgradient
(self, x)
idr/model/implicit_differentiable_renderer.py:86
↓ 2 callersFunctiongram_matrix
(y)
idr/model/loss.py:18
↓ 2 callersFunctiongram_matrix
(y)
engines/trainer.py:33
↓ 2 callersFunctionlin2img
(tensor, img_res)
idr/utils/plots.py:308
↓ 2 callersFunctionpytorch_complex_abs
(x)
utils/pytorch_projector.py:37
↓ 2 callersFunctionrender_video
Render video
engines/eval.py:144
↓ 2 callersFunctionsave_checkpoint
(path, global_step, model, optimizer)
engines/trainer.py:243
↓ 2 callersFunctionviewmatrix
(z, up, pos)
data/load_llff.py:132
↓ 1 callersMethod__init__
Nerf MLP backbone
models/renderer.py:55
↓ 1 callersFunction_load_data
(basedir, factor=None, width=None, height=None, load_imgs=True, crpt=False)
data/load_llff.py:62
↓ 1 callersFunctioncalculate_psnr
(img1, img2, mask)
idr/evaluation/eval.py:230
↓ 1 callersFunctioncompare_rotations
(R1, R2)
idr/evaluation/eval.py:224
↓ 1 callersFunctioncontrast_loss
(features, indexes)
utils/image.py:18
↓ 1 callersFunctioncreate_arg_parser
()
run_nerf.py:39
↓ 1 callersFunctioncreate_arg_parser
()
gen_dataset.py:29
↓ 1 callersMethodcreate_embedding_fn
(self)
idr/model/embedder.py:10
↓ 1 callersMethodembed
(self, inputs)
idr/model/embedder.py:35
↓ 1 callersFunctionevaluate
(**kwargs)
idr/evaluation/eval.py:17
↓ 1 callersFunctionevaluate
(**kwargs)
idr/evaluation/eval_disentanglement.py:16
↓ 1 callersMethodforward
(self, input, compute_grad=False)
idr/model/implicit_differentiable_renderer.py:67
↓ 1 callersFunctiongenerate_dataset
(args, output_path)
gen_dataset.py:63
↓ 1 callersFunctionget_3D_quiver_trace
(points, directions, color='#bd1540', name='')
idr/utils/plots.py:88
↓ 1 callersFunctionget_3D_scatter_trace
(points, name='', size=3, caption=None)
idr/utils/plots.py:67
↓ 1 callersFunctionget_Ps
(cameras,number_of_cameras)
idr/preprocess/preprocess_cameras.py:11
↓ 1 callersFunctionget_all_mask_points
(masks_dir)
idr/preprocess/preprocess_cameras.py:74
↓ 1 callersFunctionget_cameras_accuracy
Align predicted pose to gt pose and print cameras accuracy
idr/evaluation/eval.py:179
↓ 1 callersMethodget_eikonal_loss
(self, grad_theta)
idr/model/loss.py:49
↓ 1 callersFunctionget_fundamental_matrices
(P_0, Ps)
idr/preprocess/preprocess_cameras.py:67
↓ 1 callersFunctionget_fundamental_matrix
(P_1,P_2)
idr/preprocess/preprocess_cameras.py:21
↓ 1 callersFunctionget_grid
(points, resolution)
idr/utils/plots.py:242
↓ 1 callersFunctionget_idx
smooth style embedding input
engines/eval.py:56
↓ 1 callersMethodget_mask_loss
(self, sdf_output, network_object_mask, object_mask)
idr/model/loss.py:56
↓ 1 callersFunctionget_min_max_d
(curx, cury, P_j, silhouette_j, P_0, Fj0, j)
idr/preprocess/preprocess_cameras.py:41
↓ 1 callersFunctionget_normalization_function
(Ps,mask_points_all,number_of_normalization_points,number_of_cameras,masks_all)
idr/preprocess/preprocess_cameras.py:121
↓ 1 callersFunctionget_persp_intrinsic
(H, W, focal, ps=1., us=1., device=torch.device('cpu'))
utils/ray.py:48
↓ 1 callersMethodget_rbg_value
(self, points, view_dirs)
idr/model/implicit_differentiable_renderer.py:256
↓ 1 callersMethodget_rgb_loss
(self,rgb_values, rgb_gt, network_object_mask, object_mask)
idr/model/loss.py:40
↓ 1 callersMethodget_scale_mat
(self)
idr/datasets/scene_dataset.py:133
↓ 1 callersFunctionget_stl_num
(style_dir)
utils/misc.py:3
↓ 1 callersMethodget_style_loss
(self, img_res, rgb_values, rgb_gt, style_img, object_mask)
idr/model/loss.py:65
↓ 1 callersFunctionget_surface_trace
(path, epoch, sdf, resolution=100, return_mesh=False)
idr/utils/plots.py:112
↓ 1 callersFunctionimread
(f)
data/load_llff.py:112
↓ 1 callersFunctionlift
(x, y, z, intrinsics)
idr/utils/rend_util.py:87
↓ 1 callersFunctionlinear_eval
Main function of conditional style interpolation
engines/eval.py:80
↓ 1 callersFunctionload_LINEMOD_data
(basedir, half_res=False, testskip=1)
data/load_LINEMOD.py:37
↓ 1 callersFunctionload_blender_data
(basedir, half_res=False, testskip=1)
data/load_blender.py:37
↓ 1 callersFunctionload_dv_data
(scene='cube', basedir='/data/deepvoxels', testskip=8)
data/load_deepvoxels.py:6
↓ 1 callersFunctionload_llff_data
(basedir, factor=8, recenter=True, bd_factor=.75, spherify=False, path_zflat=False, crpt=False)
data/load_llff.py:247
↓ 1 callersFunctionload_pose
(filename)
data/load_deepvoxels.py:49
↓ 1 callersFunctionmain
(args)
run_nerf.py:233
↓ 1 callersFunctionmin_line_dist
(rays_o, rays_d)
data/load_llff.py:195
↓ 1 callersMethodminimal_sdf_points
Find points with minimal SDF value on rays for P_out pixels
idr/model/ray_tracing.py:273
↓ 1 callersMethodmse_dist
(self, rgb, gts)
models/vote_net.py:40
↓ 1 callersFunctionparse_intrinsics
(filepath, trgt_sidelength, invert_y=False)
data/load_deepvoxels.py:9
↓ 1 callersFunctionplot_depth_maps
(depth_maps, path, epoch, plot_nrow, img_res)
idr/utils/plots.py:275
↓ 1 callersFunctionplot_images
(rgb_points, ground_true, path, epoch, plot_nrow, img_res)
idr/utils/plots.py:289
↓ 1 callersFunctionpolar_to_rotmat
(azimuths, zeniths)
models/vote_net.py:15
↓ 1 callersFunctionpolar_to_xyz
(azimuths, zeniths, rad=1.)
models/vote_net.py:27
↓ 1 callersFunctionpose_spherical
(theta, phi, radius)
data/load_LINEMOD.py:29
↓ 1 callersFunctionpose_spherical
(theta, phi, radius)
data/load_blender.py:29
↓ 1 callersFunctionpytorch_complex_expi
(x)
utils/pytorch_projector.py:31
↓ 1 callersFunctionpytorch_complex_mul
(x1, x2)
utils/pytorch_projector.py:28
↓ 1 callersFunctionpytorch_fftfreq
(n)
utils/pytorch_projector.py:40
↓ 1 callersMethodray_sampler
Sample the ray in a given range and run secant on rays which have sign transition
idr/model/ray_tracing.py:191
↓ 1 callersFunctionrecenter_poses
(poses)
data/load_llff.py:170
↓ 1 callersFunctionrefine_visual_hull
(masks, Ps, scale, center)
idr/preprocess/preprocess_cameras.py:88
↓ 1 callersFunctionrender_path_spiral
(c2w, up, rads, focal, zdelta, zrate, rots, N)
data/load_llff.py:157
↓ 1 callersMethodrender_rays
Volumetric rendering. Args: ray_o: origins of rays. [N_rays, 3] ray_d: directions of rays. [N_rays, 3] near: the
models/nerf_net.py:88
next →1–100 of 223, ranked by callers