MCPcopy Create free account

hub / github.com/chih-hao-lin/urbanir / functions

Functions237 in github.com/chih-hao-lin/urbanir

↓ 20 callersFunctionmake_float3
models/csrc/include/helper_math.h:96
↓ 15 callersFunctionmin
models/csrc/include/helper_math.h:73
↓ 15 callersFunctionsave_image
(image, path)
utils.py:85
↓ 13 callersFunctionread_next_bytes
Read and unpack the next bytes from a binary file. :param fid: :param num_bytes: Sum of combination of {2, 4, 8}, e.g. 2, 6, 16, 30, etc.
datasets/colmap_utils.py:70
↓ 11 callersFunctionload_ckpt
(model, ckpt_path, model_name='model', prefixes_to_ignore=[])
utils.py:25
↓ 9 callersFunctionget_rays
Get ray origin and directions in world coordinate for all pixels in one image. Reference: https://www.scratchapixel.com/lessons/3d-basic-rend
datasets/ray_utils.py:50
↓ 8 callersFunctionmax
models/csrc/include/helper_math.h:68
↓ 6 callersFunctionbox_filter
Input image: (h, w) r: constant Return image: (h, w)
utils.py:45
↓ 6 callersMethoddensity
Inputs: x: (N, 3) xyz in [-scale, scale] return_feat: whether to return intermediate feature Outputs:
models/networks.py:151
↓ 5 callersFunctionnormalize
Normalize a vector.
datasets/ray_utils.py:107
↓ 5 callersFunctionparse_calib_file
(path, key)
datasets/kitti360.py:314
↓ 5 callersFunctionrender
Render rays by 1. Compute the intersection of the rays with the scene bounding box 2. Follow the process in @render_func (different for t
models/rendering.py:13
↓ 4 callersFunctionconvert_normal
(normal, pose_c2w)
utils.py:93
↓ 4 callersFunctionget_opts
()
opt.py:4
↓ 4 callersMethodget_path_rays
(self, c2w_list)
datasets/tnt.py:294
↓ 4 callersFunctionget_ray_directions
Get ray directions for all pixels in camera coordinate [right down front]. Reference: https://www.scratchapixel.com/lessons/3d-basic-renderin
datasets/ray_utils.py:8
↓ 4 callersFunctionget_sample_index
( pixel_h, pixel_w, image_h, image_w )
utility/triangulate.py:7
↓ 4 callersFunctionnormalized_tensor
(x, device='cuda')
models/shading.py:31
↓ 4 callersMethodread_meta
(self, split, imgs, c2w_list, semantics, classes=7)
datasets/tnt.py:201
↓ 3 callersMethodadd_flares
(self, rgb, **kwargs)
models/shading.py:221
↓ 3 callersFunctiondot
models/csrc/include/helper_math.h:306
↓ 3 callersFunctiongenerate_interpolated_path
Creates a smooth spline path between input keyframe camera poses. Adapted from https://github.com/google-research/multinerf/blob/main/internal/cam
datasets/ray_utils.py:230
↓ 3 callersFunctionmake_float2
models/csrc/include/helper_math.h:88
↓ 3 callersFunctionread_image
(img_path, img_wh)
datasets/color_utils.py:19
↓ 3 callersFunctionread_video_frames
(video_path)
utility/video.py:29
↓ 2 callersMethod__init__
( self, config, device='cuda' )
models/shading.py:270
↓ 2 callersFunctionaverage_poses
Calculate the average pose, which is then used to center all poses using @center_poses. Its computation is as follows: 1. Compute the cen
datasets/ray_utils.py:112
↓ 2 callersFunctiondepth2img
(depth, scale=16)
render.py:19
↓ 2 callersMethodestimate_visibility_test
( self, model, points, **kwargs )
models/shading.py:338
↓ 2 callersMethodforward_test
Inputs: x: (N, 3) xyz in [-scale, scale] d: (N, 3) directions Outputs: sigmas: (N) r
models/networks.py:227
↓ 2 callersMethodget_all_cells
Get all cells from the density grid. Outputs: cells: list (of length self.cascades) of indices and coords
models/networks.py:270
↓ 2 callersFunctionget_light_model
(config_path)
models/shading.py:15
↓ 2 callersFunctionget_mask_from_label
(label, ids)
utils.py:98
↓ 2 callersMethodget_tracing_rays
( self, model, points )
models/shading.py:323
↓ 2 callersMethodgrad
(self, x)
models/networks.py:173
↓ 2 callersFunctionread_cameras_binary
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCameras
datasets/colmap_utils.py:108
↓ 2 callersMethodread_depth
(self, depths)
datasets/tnt.py:250
↓ 2 callersMethodread_depth
(self, dir_depth, frame_id)
datasets/kitti360.py:226
↓ 2 callersFunctionread_images_binary
see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBi
datasets/colmap_utils.py:168
↓ 2 callersMethodread_normal
(self, dir_normal, frame_id)
datasets/kitti360.py:204
↓ 2 callersFunctionread_points3d_binary
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints
datasets/colmap_utils.py:230
↓ 2 callersMethodread_rgb
(self, dir_rgb, frame_id)
datasets/kitti360.py:152
↓ 2 callersFunctionread_semantic
(sem_path, sem_wh, classes=7)
datasets/color_utils.py:65
↓ 2 callersMethodread_semantics
(self, dir_sem, frame_id)
datasets/kitti360.py:163
↓ 2 callersMethodread_shadow
(self, dir_shadow, frame_id)
datasets/kitti360.py:216
↓ 2 callersFunctionrender_chunks
(model, rays_o, rays_d, chunk_size, **kwargs)
render.py:35
↓ 2 callersFunctionrender_for_test
(hparams, split='test')
render.py:52
↓ 2 callersMethodsample_index
(self, index)
utils.py:129
↓ 2 callersFunctionsample_visibility_test
( model, rays_o, rays_d, hits_t, light_dist, # Other parameters **kwargs )
models/visibility.py:6
↓ 2 callersFunctionsemantic2img
(sem_label, classes)
render.py:27
↓ 2 callersMethodshading_model
( self, model, points, albedo, normal, semantic, view_
models/shading.py:289
↓ 2 callersFunctionslim_ckpt
(ckpt_path, save_poses=False)
utils.py:33
↓ 2 callersFunctionvolume_render
( model, rays_o, rays_d, hits_t, # Other parameters **kwargs )
models/volume_render.py:6
↓ 1 callersMethod__init__
(self, hparams)
losses.py:75
↓ 1 callersMethod_init
()
models/global_var.py:7
↓ 1 callersFunctionassoc_legendre_coeff
Compute associated Legendre polynomial coefficients. Returns the coefficient of the cos^k(theta)*sin^m(theta) term in the (l, m)th associated Leg
models/ref_util.py:25
↓ 1 callersFunctionaxisangle_to_R
Convert an axis-angle vector to rotation matrix from https://github.com/ActiveVisionLab/nerfmm/blob/main/utils/lie_group_helper.py#L47 I
datasets/ray_utils.py:78
↓ 1 callersMethodchange_direction
(self, direction)
models/shading.py:286
↓ 1 callersFunctionclamp
models/csrc/include/helper_math.h:280
↓ 1 callersFunctioncompute_scale_and_shift
(prediction, target)
losses.py:10
↓ 1 callersFunctionconvert_samples_to_ply
Convert sdf samples to .ply :param pytorch_3d_tensor: a torch.FloatTensor of shape (n,n,n) :voxel_grid_origin: a list of three floats: th
extract_mesh.py:13
↓ 1 callersFunctioncreate_spheric_poses
Create circular poses around z axis. Inputs: radius: the (negative) height and the radius of the circle. mean_h: mean camera
datasets/ray_utils.py:184
↓ 1 callersMethodestimate_sunlight
(self, sun_dir)
datasets/waymo.py:188
↓ 1 callersMethodestimate_sunlight
(self, dir_oxts, dir_calib, frame_id, cam2world)
datasets/kitti360.py:242
↓ 1 callersMethodestimate_visibility_train
( self, model, points, **kwargs )
models/shading.py:348
↓ 1 callersFunctionextract_model_state_dict
(ckpt_path, model_name='model', prefixes_to_ignore=[])
utils.py:8
↓ 1 callersFunctionfmaxf
models/csrc/include/helper_math.h:63
↓ 1 callersFunctionfminf
models/csrc/include/helper_math.h:58
↓ 1 callersFunctiongeneralized_binomial_coeff
Compute generalized binomial coefficients.
models/ref_util.py:21
↓ 1 callersFunctionget_ml_array
Create a list with all pairs of (l, m) values to use in the encoding.
models/ref_util.py:50
↓ 1 callersMethodget_path_rays
(self, c2w_list)
datasets/colmap.py:239
↓ 1 callersMethodget_path_rays
(self, render_c2w)
datasets/waymo.py:109
↓ 1 callersMethodget_path_rays
(self, render_c2w)
datasets/kitti360.py:142
↓ 1 callersMethodget_shading_visibility
( self, model, points, normal, semantic, view_d, **kw
models/shading.py:181
↓ 1 callersMethodget_tracing_rays
( self, position, model, points )
models/shading.py:165
↓ 1 callersFunctioninterp
Runs multidimensional B-spline interpolation on the input points.
datasets/ray_utils.py:261
↓ 1 callersFunctionis_image_name
(name)
utility/video.py:43
↓ 1 callersMethodlabel_mapping
(self, label)
datasets/kitti360.py:173
↓ 1 callersFunctionmain
()
utility/triangulate.py:14
↓ 1 callersFunctionmain
()
utility/slim_ckpt.py:5
↓ 1 callersFunctionmerge_video
()
utility/video.py:99
↓ 1 callersFunctionnormalize
Normalize a vector.
datasets/tnt.py:15
↓ 1 callersFunctionnormalize
Normalize a vector.
datasets/colmap.py:14
↓ 1 callersMethodplace_at_camera
( self, pose_c2w )
models/shading.py:154
↓ 1 callersFunctionpoints_to_poses
Converts from (position, lookat, up) format to pose matrices.
datasets/ray_utils.py:257
↓ 1 callersFunctionposes_to_points
Converts from pose matrices to (position, lookat, up) format.
datasets/ray_utils.py:250
↓ 1 callersFunctionqvec2rotmat
(qvec)
datasets/colmap_utils.py:272
↓ 1 callersMethodqvec2rotmat
(self)
datasets/colmap_utils.py:49
↓ 1 callersFunctionread_cameras_text
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasText(const std::string& path) void Reconstruction::ReadCamerasTe
datasets/colmap_utils.py:82
↓ 1 callersMethodread_depth
(self, dir_depth, frame_id)
datasets/waymo.py:133
↓ 1 callersFunctionread_images_text
see: src/base/reconstruction.cc void Reconstruction::ReadImagesText(const std::string& path) void Reconstruction::WriteImagesText
datasets/colmap_utils.py:137
↓ 1 callersMethodread_intrinsics
(self, **kwargs)
datasets/colmap.py:56
↓ 1 callersMethodread_meta
(self, split, **kwargs)
datasets/colmap.py:79
↓ 1 callersMethodread_normal
(self, dir_normal, frame_id)
datasets/waymo.py:162
↓ 1 callersMethodread_normals
(self, dir_normal, prefix)
datasets/tnt.py:257
↓ 1 callersFunctionread_points3D_text
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3D
datasets/colmap_utils.py:203
↓ 1 callersMethodread_rgb
(self, dir_rgb, prefix)
datasets/tnt.py:282
↓ 1 callersMethodread_rgb
(self, dir_rgb, frame_id)
datasets/waymo.py:120
↓ 1 callersMethodread_semantics
(self, dir_sem, frame_id)
datasets/waymo.py:150
↓ 1 callersMethodread_shadow
(self, dir_shadow, prefix)
datasets/tnt.py:271
next →1–100 of 237, ranked by callers