MCPcopy Create free account

hub / github.com/cvlab-kaist/PF3plat / functions

Functions843 in github.com/cvlab-kaist/PF3plat

Methodget_current_decay
(self, *args, **kwargs)
src/model/unidepth/utils/ema_torch.py:25
Methodget_data_shim
(self)
src/model/encoder/encoder_costvolume.py:579
Functionget_depth
Calculate the depths that correspond to the specified 2D points on the epipolar lines defined by the origins and directions. The extrinsics and in
src/geometry/epipolar_lines.py:280
Functionget_dist_info
()
src/model/unidepth/utils/distributed.py:47
Functionget_gt_correspondence_mask
Computes the mask of valid flows (that do not match to a pixel outside of the image).
src/flow_util.py:1143
Functionget_num_layer_for_convnext_single
Each layer is assigned distinctive layer ids
src/model/unidepth/backbones/convnext2.py:7
Functionget_params
(module, lr, wd)
src/model/unidepth/utils/misc.py:72
Methodget_params
(self, lr, wd, ld, *args, **kwargs)
src/model/unidepth/backbones/dinov2.py:336
Methodget_params
(self, lr, wd, ld, *args, **kwargs)
src/model/unidepth/backbones/convnext.py:509
Methodget_params
(self, lr, wd, ld, *args, **kwargs)
src/model/unidepth/backbones/convnext2.py:275
Functionget_params_layerdecayswin
(module, lr, wd, ld)
src/model/unidepth/utils/misc.py:135
Functionget_patches
( tensor: torch.Tensor, required_corners: torch.Tensor, ps: int )
src/model/LightGlue/lightglue/aliked.py:48
Functionget_pointcloud_from_rgbd
( image, depth, mask, intrinsic_matrix, extrinsic_matrix = None, )
src/model/unidepth/utils/visualization.py:57
Functionget_seed
()
src/global_cfg.py:18
Methodglobal_step
(self)
src/dataset/view_sampler/view_sampler.py:60
Methodgroup_matcher
(self, coarse=False)
src/model/unidepth/backbones/convnext.py:484
Methodhook
(module, input, output)
src/model/encoder/visualization/encoder_visualizer_costvolume.py:48
Functionidentity
(t, *args, **kwargs)
src/model/unidepth/utils/misc.py:174
Methodindex
(self)
src/dataset/dataset_dl3dv.py:273
Methodindex
(self)
src/dataset/dataset_dl3dv_test.py:276
Methodindex
(self)
src/dataset/dataset_re10k_test.py:158
Methodindex
(self)
src/dataset/dataset_acid_test.py:155
Functioninit_
(tensor)
src/model/encoder/costvolume/ldm_unet/attention.py:27
Functioninit_weights_vit_timm
ViT weight initialization, original timm impl (for reproducibility)
src/model/unidepth/backbones/dinov2.py:352
Functioninlier_counting_3d
(X0, X1, R, t, th=0.05)
src/flow_util.py:744
Functioninner
(x)
src/model/unidepth/utils/misc.py:196
Methodinteraction
(self, *args, **kwargs)
src/dataset/dataset_re10k.py:30
Methodinteraction
(self, *args, **kwargs)
src/dataset/dataset_dl3dv.py:30
Methodinteraction
(self, *args, **kwargs)
src/dataset/dataset_dl3dv_test.py:30
Functionis_rank_zero
(args)
src/model/unidepth/utils/distributed.py:43
Functionl2norm
(t)
src/model/unidepth/utils/misc.py:166
Functionlast_stack
(tensors)
src/model/unidepth/utils/misc.py:20
Functionlinear
Create a linear module.
src/model/encoder/costvolume/ldm_unet/util.py:253
Functionload_pretrained
(state_dict, checkpoint)
src/model/unidepth/utils/misc.py:231
Methodload_pretrained
(self, model_file)
src/model/unidepth/unidepthv2/unidepthv2.py:321
Methodload_state_dict
(self, *args, **kwargs)
src/model/unidepth/utils/ema_torch.py:53
Functionlog
(t, eps: float = 1e-5)
src/model/unidepth/utils/misc.py:162
Methodlog_hyperparams
(self, params)
src/misc/LocalLogger.py:27
Methodlog_metrics
(self, metrics, step)
src/misc/LocalLogger.py:31
Functionlog_train_artifacts
(rgbs, gts, preds, ds_name, step, infos={})
src/model/unidepth/utils/visualization.py:136
Functionmake_beta_schedule
(schedule, n_timestep, linear_start=1e-4, linear_end=2e-2, cosine_s=8e-3)
src/model/encoder/costvolume/ldm_unet/util.py:21
Functionmake_ddim_sampling_parameters
(alphacums, ddim_timesteps, eta, verbose=True)
src/model/encoder/costvolume/ldm_unet/util.py:63
Functionmake_ddim_timesteps
(ddim_discr_method, num_ddim_timesteps, num_ddpm_timesteps, verbose=True)
src/model/encoder/costvolume/ldm_unet/util.py:46
Functionmake_grid
(img)
src/flow_util.py:328
Functionmake_matching_plot_fast
(image0, image1, kpts0, kpts1, mkpts0, mkpts1, color, text, path=None,
src/flow_util.py:370
Functionmatches_from_flow
Retrieves the pixel coordinates of 'good' matches in source and target images, based on provided flow field (relating the target to the sourc
src/flow_util.py:751
Functionmatches_from_flow_batch
Retrieves the pixel coordinates of 'good' matches in source and target images, based on provided flow field (relating the target to the sourc
src/flow_util.py:795
Functionmax_neg_value
(t)
src/model/encoder/costvolume/ldm_unet/attention.py:23
Functionmax_stack
(tensors)
src/model/unidepth/utils/misc.py:14
Functionmaybe
(fn)
src/model/unidepth/utils/misc.py:194
Functionmean_flat
Take the mean over all non-batch dimensions.
src/model/encoder/costvolume/ldm_unet/util.py:192
Functionmean_stack
(tensors)
src/model/unidepth/utils/misc.py:34
Functionmerge_splits
(splits, num_splits=2, channel_last=False, )
src/flow_util.py:61
Methodname
(self)
src/misc/LocalLogger.py:19
Functionnoise_like
(shape, device, repeat=False)
src/model/encoder/costvolume/ldm_unet/util.py:286
Methodnum_context_views
(self)
src/dataset/view_sampler/view_sampler_arbitrary.py:65
Methodnum_context_views
(self)
src/dataset/view_sampler/view_sampler.py:56
Methodnum_context_views
(self)
src/dataset/view_sampler/view_sampler_bounded.py:116
Methodnum_context_views
(self)
src/dataset/view_sampler/view_sampler_all.py:32
Methodnum_context_views
(self)
src/dataset/view_sampler/view_sampler_evaluation.py:62
Methodnum_target_views
(self)
src/dataset/view_sampler/view_sampler_arbitrary.py:69
Methodnum_target_views
(self)
src/dataset/view_sampler/view_sampler.py:51
Methodnum_target_views
(self)
src/dataset/view_sampler/view_sampler_bounded.py:120
Methodnum_target_views
(self)
src/dataset/view_sampler/view_sampler_all.py:36
Methodnum_target_views
(self)
src/dataset/view_sampler/view_sampler_evaluation.py:66
Methodon_test_end
(self)
src/model/model_wrapper.py:390
Functiononce
(fn)
src/model/unidepth/utils/misc.py:204
Methodoptimizer_step
( self, epoch=None, batch_idx=None, optimizer=None, optimizer_closure=
src/model/model_wrapper.py:210
Functionplot_images
Plot a set of images horizontally. Args: imgs: list of NumPy RGB (H, W, 3) or PyTorch RGB (3, H, W) or mono (H, W). titles: a list
src/model/LightGlue/lightglue/viz2d.py:42
Functionplot_keypoints
Plot keypoints for existing images. Args: kpts: list of ndarrays of size (N, 2). colors: string, or list of list of tuples (one fo
src/model/LightGlue/lightglue/viz2d.py:89
Functionplot_matches
Plot matches for a pair of existing images. Args: kpts0, kpts1: corresponding keypoints of size (N, 2). color: color of each match
src/model/LightGlue/lightglue/viz2d.py:108
Functionproject
( points: Float[Tensor, "*#batch dim"], extrinsics: Float[Tensor, "*#batch dim+1 dim+1"], intrinsi
src/geometry/projection.py:59
Functionproject_points
( points_3d, intrinsic_matrix, image_shape, )
src/model/unidepth/utils/geometric.py:158
Functionrandom_se3
()
src/flow_util.py:129
Functionrelative_disparity_to_depth
Convert relative disparity, where 0 is near and 1 is far, to depth.
src/model/encoder/costvolume/conversions.py:5
Functionremove_leading_dim
(infos)
src/model/unidepth/utils/misc.py:405
Functionremove_padding_metas
(out, image_metas)
src/model/unidepth/utils/misc.py:369
Functionresize
( image: Float[Tensor, "channel height width"], shape: Optional[tuple[int, int]] = None, width: Op
src/visualization/layout.py:208
Methodrestore
(self, *args, **kwargs)
src/model/unidepth/utils/ema_torch.py:37
Functionrsh_cart_0
Computes all real spherical harmonics up to degree 0. This is an autogenerated method. See https://github.com/cheind/torch-spherical-harmonic
src/model/unidepth/utils/sht.py:11
Functionrsh_cart_1
Computes all real spherical harmonics up to degree 1. This is an autogenerated method. See https://github.com/cheind/torch-spherical-harmonic
src/model/unidepth/utils/sht.py:36
Functionrsh_cart_2
Computes all real spherical harmonics up to degree 2. This is an autogenerated method. See https://github.com/cheind/torch-spherical-harmonic
src/model/unidepth/utils/sht.py:67
Functionrsh_cart_3
Computes all real spherical harmonics up to degree 3. This is an autogenerated method. See https://github.com/cheind/torch-spherical-harmonic
src/model/unidepth/utils/sht.py:110
Functionrsh_cart_4
Computes all real spherical harmonics up to degree 4. This is an autogenerated method. See https://github.com/cheind/torch-spherical-harmonic
src/model/unidepth/utils/sht.py:160
Functionrsh_cart_5
Computes all real spherical harmonics up to degree 5. This is an autogenerated method. See https://github.com/cheind/torch-spherical-harmonic
src/model/unidepth/utils/sht.py:228
Functionrsh_cart_6
Computes all real spherical harmonics up to degree 6. This is an autogenerated method. See https://github.com/cheind/torch-spherical-harmonic
src/model/unidepth/utils/sht.py:329
Functionrsh_cart_7
Computes all real spherical harmonics up to degree 7. This is an autogenerated method. See https://github.com/cheind/torch-spherical-harmonic
src/model/unidepth/utils/sht.py:509
Functionrsh_cart_8
Computes all real spherical harmonics up to degree 8. This is an autogenerated method. See https://github.com/cheind/torch-spherical-harmonic
src/model/unidepth/utils/sht.py:833
Methodsample
Arbitrarily sample context and target views.
src/dataset/view_sampler/view_sampler_arbitrary.py:21
Methodsample
( self, scene: str, extrinsics: Float[Tensor, "view 4 4"], intrinsics: Float[T
src/dataset/view_sampler/view_sampler_bounded.py:29
Methodsample
( self, scene: str, extrinsics: Float[Tensor, "view 4 4"], intrinsics: Float[T
src/dataset/view_sampler/view_sampler_all.py:17
Methodsample
( self, scene: str, extrinsics: Float[Tensor, "view 4 4"], intrinsics: Float[T
src/dataset/view_sampler/view_sampler_evaluation.py:44
Functionsample_training_rays
( image: Float[Tensor, "batch view channel ..."], intrinsics: Float[Tensor, "batch view dim dim"],
src/geometry/projection.py:140
Functionsave_file_ply
(xyz, rgb, pc_file)
src/model/unidepth/utils/visualization.py:107
Functionsave_plot
Save the current figure without any white margin.
src/model/LightGlue/lightglue/viz2d.py:192
Functionscale_module
Scale the parameters of a module and return it.
src/model/encoder/costvolume/ldm_unet/util.py:183
Functionselect_cameras
Camera Parameters - c2w : (batch, num_views, 4, 4)
src/flow_util.py:1208
Functionselect_gaussians
Gaussian Parameters - covariances : (batch, num_gaussians, 3, 3) - harmonics : (batch, num_gaussians, 3, 1) - means : (batch, num_gau
src/flow_util.py:1187
Functionseparate_loss_cfg_wrappers
(joined: dict)
src/config.py:73
Methodset_original_shapes
(self, shapes: tuple[int, int])
src/model/unidepth/unidepthv2/decoder.py:124
← previousnext →701–800 of 843, ranked by callers