Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/chih-hao-lin/urbanir
/ functions
Functions
237 in github.com/chih-hao-lin/urbanir
⨍
Functions
237
◇
Types & classes
27
↓ 20 callers
Function
make_float3
models/csrc/include/helper_math.h:96
↓ 15 callers
Function
min
models/csrc/include/helper_math.h:73
↓ 15 callers
Function
save_image
(image, path)
utils.py:85
↓ 13 callers
Function
read_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 callers
Function
load_ckpt
(model, ckpt_path, model_name='model', prefixes_to_ignore=[])
utils.py:25
↓ 9 callers
Function
get_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 callers
Function
max
models/csrc/include/helper_math.h:68
↓ 6 callers
Function
box_filter
Input image: (h, w) r: constant Return image: (h, w)
utils.py:45
↓ 6 callers
Method
density
Inputs: x: (N, 3) xyz in [-scale, scale] return_feat: whether to return intermediate feature Outputs:
models/networks.py:151
↓ 5 callers
Function
normalize
Normalize a vector.
datasets/ray_utils.py:107
↓ 5 callers
Function
parse_calib_file
(path, key)
datasets/kitti360.py:314
↓ 5 callers
Function
render
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 callers
Function
convert_normal
(normal, pose_c2w)
utils.py:93
↓ 4 callers
Function
get_opts
()
opt.py:4
↓ 4 callers
Method
get_path_rays
(self, c2w_list)
datasets/tnt.py:294
↓ 4 callers
Function
get_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 callers
Function
get_sample_index
( pixel_h, pixel_w, image_h, image_w )
utility/triangulate.py:7
↓ 4 callers
Function
normalized_tensor
(x, device='cuda')
models/shading.py:31
↓ 4 callers
Method
read_meta
(self, split, imgs, c2w_list, semantics, classes=7)
datasets/tnt.py:201
↓ 3 callers
Method
add_flares
(self, rgb, **kwargs)
models/shading.py:221
↓ 3 callers
Function
dot
models/csrc/include/helper_math.h:306
↓ 3 callers
Function
generate_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 callers
Function
make_float2
models/csrc/include/helper_math.h:88
↓ 3 callers
Function
read_image
(img_path, img_wh)
datasets/color_utils.py:19
↓ 3 callers
Function
read_video_frames
(video_path)
utility/video.py:29
↓ 2 callers
Method
__init__
( self, config, device='cuda' )
models/shading.py:270
↓ 2 callers
Function
average_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 callers
Function
depth2img
(depth, scale=16)
render.py:19
↓ 2 callers
Method
estimate_visibility_test
( self, model, points, **kwargs )
models/shading.py:338
↓ 2 callers
Method
forward_test
Inputs: x: (N, 3) xyz in [-scale, scale] d: (N, 3) directions Outputs: sigmas: (N) r
models/networks.py:227
↓ 2 callers
Method
get_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 callers
Function
get_light_model
(config_path)
models/shading.py:15
↓ 2 callers
Function
get_mask_from_label
(label, ids)
utils.py:98
↓ 2 callers
Method
get_tracing_rays
( self, model, points )
models/shading.py:323
↓ 2 callers
Method
grad
(self, x)
models/networks.py:173
↓ 2 callers
Function
read_cameras_binary
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCameras
datasets/colmap_utils.py:108
↓ 2 callers
Method
read_depth
(self, depths)
datasets/tnt.py:250
↓ 2 callers
Method
read_depth
(self, dir_depth, frame_id)
datasets/kitti360.py:226
↓ 2 callers
Function
read_images_binary
see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBi
datasets/colmap_utils.py:168
↓ 2 callers
Method
read_normal
(self, dir_normal, frame_id)
datasets/kitti360.py:204
↓ 2 callers
Function
read_points3d_binary
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints
datasets/colmap_utils.py:230
↓ 2 callers
Method
read_rgb
(self, dir_rgb, frame_id)
datasets/kitti360.py:152
↓ 2 callers
Function
read_semantic
(sem_path, sem_wh, classes=7)
datasets/color_utils.py:65
↓ 2 callers
Method
read_semantics
(self, dir_sem, frame_id)
datasets/kitti360.py:163
↓ 2 callers
Method
read_shadow
(self, dir_shadow, frame_id)
datasets/kitti360.py:216
↓ 2 callers
Function
render_chunks
(model, rays_o, rays_d, chunk_size, **kwargs)
render.py:35
↓ 2 callers
Function
render_for_test
(hparams, split='test')
render.py:52
↓ 2 callers
Method
sample_index
(self, index)
utils.py:129
↓ 2 callers
Function
sample_visibility_test
( model, rays_o, rays_d, hits_t, light_dist, # Other parameters **kwargs )
models/visibility.py:6
↓ 2 callers
Function
semantic2img
(sem_label, classes)
render.py:27
↓ 2 callers
Method
shading_model
( self, model, points, albedo, normal, semantic, view_
models/shading.py:289
↓ 2 callers
Function
slim_ckpt
(ckpt_path, save_poses=False)
utils.py:33
↓ 2 callers
Function
volume_render
( model, rays_o, rays_d, hits_t, # Other parameters **kwargs )
models/volume_render.py:6
↓ 1 callers
Method
__init__
(self, hparams)
losses.py:75
↓ 1 callers
Method
_init
()
models/global_var.py:7
↓ 1 callers
Function
assoc_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 callers
Function
axisangle_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 callers
Method
change_direction
(self, direction)
models/shading.py:286
↓ 1 callers
Function
clamp
models/csrc/include/helper_math.h:280
↓ 1 callers
Function
compute_scale_and_shift
(prediction, target)
losses.py:10
↓ 1 callers
Function
convert_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 callers
Function
create_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 callers
Method
estimate_sunlight
(self, sun_dir)
datasets/waymo.py:188
↓ 1 callers
Method
estimate_sunlight
(self, dir_oxts, dir_calib, frame_id, cam2world)
datasets/kitti360.py:242
↓ 1 callers
Method
estimate_visibility_train
( self, model, points, **kwargs )
models/shading.py:348
↓ 1 callers
Function
extract_model_state_dict
(ckpt_path, model_name='model', prefixes_to_ignore=[])
utils.py:8
↓ 1 callers
Function
fmaxf
models/csrc/include/helper_math.h:63
↓ 1 callers
Function
fminf
models/csrc/include/helper_math.h:58
↓ 1 callers
Function
generalized_binomial_coeff
Compute generalized binomial coefficients.
models/ref_util.py:21
↓ 1 callers
Function
get_ml_array
Create a list with all pairs of (l, m) values to use in the encoding.
models/ref_util.py:50
↓ 1 callers
Method
get_path_rays
(self, c2w_list)
datasets/colmap.py:239
↓ 1 callers
Method
get_path_rays
(self, render_c2w)
datasets/waymo.py:109
↓ 1 callers
Method
get_path_rays
(self, render_c2w)
datasets/kitti360.py:142
↓ 1 callers
Method
get_shading_visibility
( self, model, points, normal, semantic, view_d, **kw
models/shading.py:181
↓ 1 callers
Method
get_tracing_rays
( self, position, model, points )
models/shading.py:165
↓ 1 callers
Function
interp
Runs multidimensional B-spline interpolation on the input points.
datasets/ray_utils.py:261
↓ 1 callers
Function
is_image_name
(name)
utility/video.py:43
↓ 1 callers
Method
label_mapping
(self, label)
datasets/kitti360.py:173
↓ 1 callers
Function
main
()
utility/triangulate.py:14
↓ 1 callers
Function
main
()
utility/slim_ckpt.py:5
↓ 1 callers
Function
merge_video
()
utility/video.py:99
↓ 1 callers
Function
normalize
Normalize a vector.
datasets/tnt.py:15
↓ 1 callers
Function
normalize
Normalize a vector.
datasets/colmap.py:14
↓ 1 callers
Method
place_at_camera
( self, pose_c2w )
models/shading.py:154
↓ 1 callers
Function
points_to_poses
Converts from (position, lookat, up) format to pose matrices.
datasets/ray_utils.py:257
↓ 1 callers
Function
poses_to_points
Converts from pose matrices to (position, lookat, up) format.
datasets/ray_utils.py:250
↓ 1 callers
Function
qvec2rotmat
(qvec)
datasets/colmap_utils.py:272
↓ 1 callers
Method
qvec2rotmat
(self)
datasets/colmap_utils.py:49
↓ 1 callers
Function
read_cameras_text
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasText(const std::string& path) void Reconstruction::ReadCamerasTe
datasets/colmap_utils.py:82
↓ 1 callers
Method
read_depth
(self, dir_depth, frame_id)
datasets/waymo.py:133
↓ 1 callers
Function
read_images_text
see: src/base/reconstruction.cc void Reconstruction::ReadImagesText(const std::string& path) void Reconstruction::WriteImagesText
datasets/colmap_utils.py:137
↓ 1 callers
Method
read_intrinsics
(self, **kwargs)
datasets/colmap.py:56
↓ 1 callers
Method
read_meta
(self, split, **kwargs)
datasets/colmap.py:79
↓ 1 callers
Method
read_normal
(self, dir_normal, frame_id)
datasets/waymo.py:162
↓ 1 callers
Method
read_normals
(self, dir_normal, prefix)
datasets/tnt.py:257
↓ 1 callers
Function
read_points3D_text
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3D
datasets/colmap_utils.py:203
↓ 1 callers
Method
read_rgb
(self, dir_rgb, prefix)
datasets/tnt.py:282
↓ 1 callers
Method
read_rgb
(self, dir_rgb, frame_id)
datasets/waymo.py:120
↓ 1 callers
Method
read_semantics
(self, dir_sem, frame_id)
datasets/waymo.py:150
↓ 1 callers
Method
read_shadow
(self, dir_shadow, prefix)
datasets/tnt.py:271
next →
1–100 of 237, ranked by callers