MCPcopy Create free account

hub / github.com/Anttwo/MILo / functions

Functions505 in github.com/Anttwo/MILo

↓ 2 callersFunctionmake_dir
(path)
milo/eval/tnt/util.py:4
↓ 2 callersMethodoneupSHdegree
(self)
milo/scene/gaussian_model.py:375
↓ 2 callersFunctionpsnr
(img1, img2)
milo/utils/image_utils.py:17
↓ 2 callersFunctionqvec2rotmat
(qvec)
milo/scene/colmap_loader.py:43
↓ 2 callersFunctionreadCamerasFromTransforms
(path, transformsfile, white_background, extension=".png")
milo/scene/dataset_readers.py:172
↓ 2 callersFunctionregistration_vol_ds
( source, gt_target, init_trans, crop_volume, voxel_size, threshold, max_itr,
milo/eval/tnt/registration.py:166
↓ 2 callersFunctionrender_set
(model_path, name, iteration, views, gaussians, pipeline, background)
milo/render.py:23
↓ 2 callersFunctionrender_set
(output_path, name, views, render_with_color_field)
milo/eval/mesh_nvs/render_mesh_nvs.py:98
↓ 2 callersMethodreplace_tensor_to_optimizer
(self, tensor, name)
milo/scene/gaussian_model.py:743
↓ 2 callersFunctionreset_mesh_state_at_next_iteration
(mesh_state)
milo/regularization/regularizer/mesh.py:623
↓ 2 callersMethodreset_occupancy
(self, base_occupancy, gaussian_idx=None, occupancy=None)
milo/scene/gaussian_model.py:278
↓ 2 callersMethodrestore
(self, model_args, training_args)
milo/scene/gaussian_model.py:138
↓ 2 callersMethodrun
A command's raison d'etre: carry out the action it exists to perform, controlled by the options initialized in 'initialize_options()',
submodules/tetra_triangulation/setup.py:54
↓ 2 callersFunctionssim
(img1, img2, window_size=11, size_average=True)
milo/utils/loss_utils.py:55
↓ 2 callersFunctionstorePly
(path, xyz, rgb)
milo/scene/dataset_readers.py:115
↓ 2 callersFunctiontrajectory_alignment
(map_file, traj_to_register, gt_traj_col, gt_trans, scene)
milo/eval/tnt/registration.py:66
↓ 2 callersFunctiontransform_points_to_pixel_space
Transform points from world space (3 coordinates) to pixel space (2 coordinates). Args: points (torch.Tensor): Should have shape
milo/utils/geometry_utils.py:82
↓ 2 callersFunctiontransform_points_view_to_world
Transform points from view space to world space. Args: points (torch.Tensor): Should have shape (n_cameras, N, 3). cameras (List[
milo/utils/geometry_utils.py:57
↓ 2 callersFunctionunflatten_voronoi_features
Unflatten the voronoi features into a 3D tensor with shape (n_gaussians, n_voronoi_per_gaussians, *voronoi_features.shape[1:]). Args: vor
milo/utils/geometry_utils.py:211
↓ 2 callersMethodupdate_learning_rate
Learning rate scheduling per step
milo/scene/gaussian_model.py:577
↓ 2 callersFunctionviewmatrix
(z, up, pos)
milo/utils/video_utils.py:130
↓ 2 callersFunctionwrite_color_distances
(path, pcd, distances, max_distance)
milo/eval/tnt/evaluation.py:50
↓ 2 callersFunctionwrite_vis_pcd
(file, points, colors)
milo/eval/dtu/eval.py:21
↓ 1 callersFunctionEvaluateHisto
( source, target, trans, crop_volume, voxel_size, threshold, filename_mvs, plo
milo/eval/tnt/evaluation.py:60
↓ 1 callersFunctionN_to_reso
(n_voxels, bbox)
milo/eval/mesh_nvs/tensorf_utils.py:14
↓ 1 callersMethodTV_loss
(self)
milo/eval/mesh_nvs/mesh_nvs_utils.py:267
↓ 1 callersMethod__init__
(self, num_input_channels, num_output_channels)
milo/scene/appearance_network.py:6
↓ 1 callersMethod__init__
(self, raster_settings)
submodules/diff-gaussian-rasterization_ms/diff_gaussian_rasterization_ms/__init__.py:360
↓ 1 callersFunction_evaluate_sdf_values
Evaluate the TSDF values at the given points using Depth Fusion. Args: points (torch.Tensor): Points at which to compute the TSDF. Has sh
milo/regularization/sdf/depth_fusion.py:400
↓ 1 callersFunction_init_cdf_mask
( importance:torch.Tensor, thres:float=1.0 )
milo/functional/func_utils.py:12
↓ 1 callersMethod_prune_optimizer
(self, mask)
milo/scene/gaussian_model.py:760
↓ 1 callersFunction_render_simp
Background tensor (bg_color) must be on GPU!
milo/functional/func_utils.py:31
↓ 1 callersFunction_sqrt_positive_part
Returns torch.sqrt(torch.max(0, x)) but with a zero subgradient where x is 0.
milo/blender/blender_utils.py:167
↓ 1 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average=True)
milo/utils/loss_utils.py:65
↓ 1 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average=True)
submodules/fused-ssim/tests/test.py:34
↓ 1 callersFunction_unbatched_marching_tetrahedra
unbatched marching tetrahedra. Refer to :func:`marching_tetrahedra`.
milo/utils/tetmesh.py:47
↓ 1 callersMethodadd_densification_stats
(self, viewspace_point_tensor, update_filter)
milo/scene/gaussian_model.py:884
↓ 1 callersMethodadd_densification_stats_culling
(self, viewspace_point_tensor, update_filter, factor)
milo/scene/gaussian_model.py:889
↓ 1 callersMethodadd_densification_stats_radegs
( self, viewspace_point_tensor, update_filter )
milo/scene/gaussian_model.py:1703
↓ 1 callersFunctionapply_depthanything
_summary_ Args: model (DepthAnythingV2): _description_ image (torch.Tensor): Has shape (H, W, 3) and RGB format. Values are betwe
milo/regularization/depth/depthanythingv2.py:30
↓ 1 callersFunctionapply_iterative_linearization_to_sdf
Starting from initial SDF values, progressively linearize the SDF values to make them match a set of isosurface vertices after applying March
milo/regularization/sdf/learnable.py:113
↓ 1 callersFunctionapply_mesh_edits_to_gaussians
( initial_mesh:Meshes, edited_mesh:Meshes, gaussian_to_vert_idx:torch.Tensor, means:torch.Tens
milo/blender/editing.py:167
↓ 1 callersFunctionapply_poses_to_meshes
( meshes:List[Meshes], i_frame:int, package:dict, )
milo/blender/editing.py:51
↓ 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
milo/eval/mesh_nvs/tensorf_utils.py:152
↓ 1 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
milo/utils/video_utils.py:41
↓ 1 callersFunctionbest_fit_transform
Calculates the least-squares best-fit transform that maps corresponding points A to B in m spatial dimensions Input: A: Nxm numpy array
milo/eval/dtu/evaluate_dtu_mesh.py:23
↓ 1 callersMethodbuild_extension
(self, ext)
submodules/tetra_triangulation/setup.py:65
↓ 1 callersFunctioncamera_to_JSON
(id, camera : Camera)
milo/utils/camera_utils.py:73
↓ 1 callersMethodcat_tensors_to_optimizer
(self, tensors_dict)
milo/scene/gaussian_model.py:807
↓ 1 callersMethodcompute_appfeature
(self, xyz_sampled)
milo/eval/mesh_nvs/mesh_nvs_utils.py:252
↓ 1 callersFunctioncompute_depth_order_loss
Compute a loss encouraging pixels in 'depth' to have the same relative depth order as in 'prior_depth'. This loss does not require prior depth map
milo/regularization/depth/depth_order.py:35
↓ 1 callersMethodconstruct_list_of_attributes
(self)
milo/scene/gaussian_model.py:586
↓ 1 callersMethodcreate_from_pcd
(self, pcd : BasicPointCloud, spatial_lr_scale : float)
milo/scene/gaussian_model.py:379
↓ 1 callersFunctioncreate_window
(window_size, channel)
milo/utils/loss_utils.py:49
↓ 1 callersFunctioncreate_window
(window_size, channel)
submodules/fused-ssim/tests/test.py:18
↓ 1 callersFunctioncull_mesh
(cameras, mesh)
milo/eval/dtu/evaluate_dtu_mesh.py:89
↓ 1 callersMethodcull_mesh
Extract mesh from scene representation and save mesh to file. Args: mesh_out_file: (str), output mesh filena
milo/eval/tnt/cull_mesh.py:205
↓ 1 callersMethodculling_with_clone
(self, scene, render_simp, iteration, args, pipe, background)
milo/scene/gaussian_model.py:1264
↓ 1 callersMethodculling_with_interesction_preserving
(self, scene, render_simp, iteration, args, pipe, background)
milo/scene/gaussian_model.py:1347
↓ 1 callersMethodculling_with_interesction_sampling
(self, scene, render_simp, iteration, args, pipe, background)
milo/scene/gaussian_model.py:1393
↓ 1 callersMethoddefine_modules
(self)
milo/eval/mesh_nvs/mesh_nvs_utils.py:226
↓ 1 callersMethoddensify_and_clone_radegs
( self, grads, grad_threshold, grads_abs, grad_abs_threshold, scene_extent )
milo/scene/gaussian_model.py:1627
↓ 1 callersMethoddensify_and_prune_mask
(self, max_grad, min_opacity, extent, max_screen_size, mask_split)
milo/scene/gaussian_model.py:967
↓ 1 callersMethoddensify_and_prune_radegs
( self, max_grad, min_opacity, extent, max_screen_size )
milo/scene/gaussian_model.py:1673
↓ 1 callersMethoddensify_and_split
(self, grads, grad_threshold, scene_extent, N=2)
milo/scene/gaussian_model.py:925
↓ 1 callersMethoddensify_and_split_mask
(self, grads, grad_threshold, scene_extent, mask, N=2)
milo/scene/gaussian_model.py:985
↓ 1 callersMethoddensify_and_split_radegs
( self, grads, grad_threshold, grads_abs, grad_abs_threshold, scene_extent, N=2
milo/scene/gaussian_model.py:1577
↓ 1 callersFunctiondepth2point_cam
(sampled_depth, ref_intrinsic)
milo/gaussian_renderer/pgsr.py:19
↓ 1 callersFunctiondepth2point_world
(depth_image, intrinsic_matrix, extrinsic_matrix)
milo/gaussian_renderer/pgsr.py:33
↓ 1 callersFunctiondepth_pcd2normal
(xyz, offset=None, gt_image=None)
milo/gaussian_renderer/pgsr.py:43
↓ 1 callersMethoddepth_reinit
(self, scene, render_depth, iteration, num_depth, args, pipe, background)
milo/scene/gaussian_model.py:1070
↓ 1 callersFunctiondepths_to_points
(view, depthmap1, depthmap2=None)
milo/utils/geometry_utils.py:135
↓ 1 callersFunctiondepths_to_points_and_rays_d
(view, depthmap1)
milo/eval/mesh_nvs/mesh_nvs_utils.py:44
↓ 1 callersMethodestimate_bounding_sphere
Estimate the bounding sphere given camera pose
milo/utils/regular_tsdf_utils.py:126
↓ 1 callersFunctioneval_sh_bases
Evaluate spherical harmonics bases at unit directions, without taking linear combination. At each point, the final result may the be
milo/eval/mesh_nvs/tensorf_utils.py:50
↓ 1 callersFunctionevaluate
(model_paths)
milo/metrics.py:36
↓ 1 callersFunctionevaluate_cull_sdf_values
( points, views, masks, gaussians, pipeline, background, kernel_size, return_colors=False, isosurface
milo/regularization/sdf/integration.py:26
↓ 1 callersFunctionevaluate_mesh
(dataset : ModelParams, iteration : int, DTU_PATH : str, ply_name : str)
milo/eval/dtu/evaluate_dtu_mesh.py:148
↓ 1 callersMethodextend_features_rest
(self)
milo/scene/gaussian_model.py:1482
↓ 1 callersFunctionextract_depth_from_mesh
Adapted from Go-Surf: https://github.com/JingwenWang95/go-surf
milo/eval/tnt/cull_mesh.py:17
↓ 1 callersFunctionextract_mesh
( dataset : ModelParams, iteration : int, pipeline : PipelineParams, n_delaunay_sites=None, mtet_on_c
milo/mesh_extract_integration.py:156
↓ 1 callersFunctionextract_mesh
(dataset, pipe, checkpoint_iterations=None, occupancy_mode="occupancy_shift")
milo/eval/dtu/mesh_extract_dtu.py:32
↓ 1 callersMethodextract_mesh_bounded
Perform TSDF fusion given a fixed depth range, used in the paper. voxel_size: the voxel size of the volume sdf_trunc
milo/utils/regular_tsdf_utils.py:140
↓ 1 callersFunctionextract_mesh_with_sdf_refinement
( dataset, iteration, pipe, n_delaunay_sites, mtet_on_cpu, refine_iter, refine
milo/mesh_extract_sdf.py:48
↓ 1 callersFunctionfix_normal_map
_summary_ Args: view (_type_): _description_ normal (_type_): Returns: _type_: _description_
milo/utils/log_utils.py:17
↓ 1 callersMethodflush
(self)
milo/utils/general_utils.py:125
↓ 1 callersFunctionfocus_point_fn
Calculate nearest point to all focal axes in poses.
milo/utils/regular_tsdf_utils.py:19
↓ 1 callersFunctionfuse_fragments
(fragments1:Fragments, fragments2:Fragments)
milo/scene/mesh.py:327
↓ 1 callersFunctiongaussian
(window_size, sigma)
milo/utils/loss_utils.py:45
↓ 1 callersFunctiongaussian
(window_size, sigma)
submodules/fused-ssim/tests/test.py:14
↓ 1 callersFunctiongen_sparse_trajectory
(mapping, f_trajectory)
milo/eval/tnt/registration.py:59
↓ 1 callersFunctiongeneralized_binomial_coeff
Compute generalized binomial coefficients.
milo/eval/mesh_nvs/tensorf_utils.py:146
↓ 1 callersFunctiongenerate_ide_fn
Generate integrated directional encoding (IDE) function. This function returns a function that computes the integrated directional encoding from
milo/eval/mesh_nvs/tensorf_utils.py:178
↓ 1 callersFunctiongetProjectionMatrix
(znear, zfar, fovX, fovY)
milo/utils/graphics_utils.py:51
↓ 1 callersMethodget_appearance_embedding
(self, idx)
milo/scene/gaussian_model.py:303
↓ 1 callersFunctionget_center_and_diag
(cam_centers)
milo/scene/dataset_readers.py:46
↓ 1 callersMethodget_connected_mesh
(self, mesh, get_largest_components=False)
milo/eval/tnt/cull_mesh.py:187
↓ 1 callersFunctionget_edited_gaussians_from_scene_state
( scene_state, i_frame, package, camera, filter_big_gaussians_with_th:float=-1., clamp
milo/blender/editing.py:355
↓ 1 callersFunctionget_edited_gaussians_parameters_from_scene_state
( scene_state, i_frame, package, return_LT:bool=False, filter_big_gaussians_with_th:float=
milo/blender/editing.py:227
↓ 1 callersFunctionget_expon_lr_func
Copied from Plenoxels Continuous learning rate decay function. Adapted from JaxNeRF The returned rate is lr_init when step=0 and lr_fina
milo/utils/general_utils.py:29
← previousnext →101–200 of 505, ranked by callers