Functions — github.com/cvg/NoPoSplat

CodeHub MCPcopy Create free account

hub / github.com/cvg/NoPoSplat / functions

Functions553 in github.com/cvg/NoPoSplat

⨍ Functions 553◇ Types & classes 135↳ Endpoints 1

Method__getitem__

(self, index: int)

src/dataset/validation_wrapper.py:25

( self, cfg: DatasetScannetPoseCfg, stage: Stage, view_sampler: ViewSampler,

src/dataset/dataset_scannet_pose.py:57

( self, dataset_cfgs: list[DatasetCfgWrapper], data_loader_cfg: DataLoaderCfg,

src/dataset/data_module.py:65

( self, cfg: DatasetRE10kCfg, stage: Stage, view_sampler: ViewSampler, )

src/dataset/dataset_re10k.py:63

(self, dataset: Dataset, length: int)

src/dataset/validation_wrapper.py:16

( self, cfg: T, stage: Stage, is_overfitting: bool, cameras_are_circul

src/dataset/view_sampler/view_sampler.py:21

( self, cfg: ViewSamplerEvaluationCfg, stage: Stage, is_overfitting: bool,

src/dataset/view_sampler/view_sampler_evaluation.py:29

( self, optimizer_cfg: OptimizerCfg, test_cfg: TestCfg, train_cfg: TrainCfg,

src/model/model_wrapper.py:99

(self, num_octaves: int)

src/model/encodings/positional_encoding.py:14

(self, cfg: EncoderNoPoSplatCfg)

src/model/encoder/encoder_noposplat.py:66

src/model/encoder/encoder.py:15

(self, cfg: EncoderNoPoSplatCfg)

src/model/encoder/encoder_noposplat_multi.py:48

(self, cfg: BackboneResnetCfg, d_in: int)

src/model/encoder/backbone/backbone_resnet.py:31

(self, cfg: BackboneCrocoCfg, d_in: int)

src/model/encoder/backbone/backbone_croco.py:67

(self, cfg: BackboneDinoCfg, d_in: int)

src/model/encoder/backbone/backbone_dino.py:22

src/model/encoder/backbone/backbone.py:15

(self, cfg: BackboneCrocoCfg, d_in: int)

src/model/encoder/backbone/backbone_croco_multiview.py:57

(self, num_patches, mask_ratio)

src/model/encoder/backbone/croco/masking.py:17

(self, drop_prob: float = 0., scale_by_keep: bool = True)

src/model/encoder/backbone/croco/blocks.py:47

(self, dim, rope=None, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0.)

src/model/encoder/backbone/croco/blocks.py:83

(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, drop=0., attn_drop=0., drop_path=0., act

src/model/encoder/backbone/croco/blocks.py:116

(self, dim, rope=None, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0.)

src/model/encoder/backbone/croco/blocks.py:134

(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, drop=0., attn_drop=0., drop_path=0., act

src/model/encoder/backbone/croco/blocks.py:173

src/model/encoder/backbone/croco/blocks.py:198

(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768, norm_layer=None, flatten=True)

src/model/encoder/backbone/croco/blocks.py:212

(self, freq=100.0, F0=1.0)

src/model/encoder/backbone/croco/pos_embed.py:114

(self, img_size=224, # input image size patch_size=16, #

src/model/encoder/backbone/croco/croco.py:23

(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768, norm_layer=None, flatten=True)

src/model/encoder/backbone/croco/patch_embed.py:38

(self, freq=100.0, F0=1.0)

src/model/encoder/backbone/croco/curope/curope2d.py:33

(self, cfg: T_cfg, encoder: T_encoder)

src/model/encoder/visualization/encoder_visualizer.py:15

(self, cfg: GaussianAdapterCfg)

src/model/encoder/common/gaussian_adapter.py:35

(self, *, n_cls_token=0, hooks_idx=None, dim_tokens=None, output_width_ratio=1, num_channels=

src/model/encoder/heads/dpt_head.py:76

(self, net, has_conf=False)

src/model/encoder/heads/linear_head.py:18

(self, *, n_cls_token=0, hooks_idx=None, dim_tokens=None, output_width_ratio=1, num_channels=

src/model/encoder/heads/dpt_gs_head.py:163

(self, in_channel=3, encoder_dim=[256, 256, 256], norm_fn='group')

src/model/encoder/heads/head_modules.py:63

(self, output_dim=[128], encoder_dim=[64, 96, 128])

src/model/encoder/heads/head_modules.py:94

Init. Args: features (int): number of features

src/model/encoder/heads/dpt_block.py:82

Init. Args: features (int): number of features

src/model/encoder/heads/dpt_block.py:147

(self, num_channels: int = 1, stride_level: int = 1, patch_

src/model/encoder/heads/dpt_block.py:280

src/model/decoder/decoder.py:30

( self, cfg: DecoderSplattingCUDACfg, )

src/model/decoder/decoder_splatting_cuda.py:25

(self, output_mode='pts3d', head_type='linear', depth_mode=

src/model/distiller/dust3d_backbone.py:26

( self, dim, heads=8, dim_head=64, dropout=0.0, selfatt=True, kv_dim=None )

src/model/transformer/attention.py:31

( self, dim, depth, heads, dim_head, mlp_dim, dropout=

src/model/transformer/transformer.py:33

(self, dim, fn)

src/model/transformer/pre_norm.py:29

(self, dim, hidden_dim, dropout=0.0)

src/model/transformer/feed_forward.py:29

src/misc/step_tracker.py:13

src/misc/LocalLogger.py:13

(self, m, n, dtype=torch.float32)

src/misc/sht.py:1414

src/misc/benchmarker.py:12

(self, cfg: EvaluationIndexGeneratorCfg)

src/evaluation/evaluation_index_generator.py:42

(self, cfg: EvaluationCfg)

src/evaluation/metric_computer.py:22

(self, cfg: EvaluationCfg, encoder, decoder, losses)

src/evaluation/pose_evaluator.py:37

(self, cfg: T_wrapper)

src/loss/loss.py:20

src/loss/loss_point.py:73

(self, criterion=None)

src/loss/loss_point.py:127

(self, pixel_loss, alpha=1)

src/loss/loss_point.py:155

(self, norm_mode='avg_dis', alpha=0.2, gt_scale=False)

src/loss/loss_point.py:199

(self, cfg: LossLpipsCfgWrapper)

src/loss/loss_lpips.py:30

r""" class for ms-ssim Args: data_range (float or int, optional): value range of input images. (usually 1.0 or 255) si

src/loss/loss_ssim.py:318

src/dataset/dataset_scannet_pose.py:82

src/dataset/dataset_re10k.py:91

src/dataset/validation_wrapper.py:22

src/loss/loss_point.py:84

src/loss/loss_point.py:100

Method_downstream_head

(self, head_num, decout, img_shape)

src/model/encoder/backbone/backbone_croco.py:216

Method_init_weights

src/model/encoder/backbone/croco/blocks.py:238

Method_set_decoder

(self, enc_embed_dim, dec_embed_dim, dec_num_heads, dec_depth, mlp_ratio, norm_layer, norm_im2_in_dec)

src/model/encoder/backbone/backbone_croco.py:96

Method_set_decoder

(self, enc_embed_dim, dec_embed_dim, dec_num_heads, dec_depth, mlp_ratio, norm_layer, norm_im2_in_dec)

src/model/encoder/backbone/backbone_croco_multiview.py:87

Method_set_patch_embed

(self, img_size=224, patch_size=16, enc_embed_dim=768, in_chans=3)

src/model/encoder/backbone/backbone_croco.py:92

Method_set_patch_embed

(self, img_size=224, patch_size=16, enc_embed_dim=768, in_chans=3)

src/model/encoder/backbone/backbone_croco_multiview.py:83

Method_set_patch_embed

(self, img_size=224, patch_size=16, enc_embed_dim=768)

src/model/distiller/dust3d_backbone.py:44

Method_set_prediction_head

No prediction head

src/model/encoder/backbone/backbone_croco.py:128

Method_set_prediction_head

No prediction head

src/model/encoder/backbone/backbone_croco_multiview.py:119

Method_set_prediction_head

No prediction head

src/model/distiller/dust3d_backbone.py:65

Functionadapt_head_conv

src/misc/weight_modify.py:117

Functionapply_bounds_shim

Compute reasonable near and far planes (lower and upper bounds on depth). This assumes that all of an example's views are of roughly the same thin

src/dataset/shims/bounds_shim.py:40

Functionapply_color_map_2d

( x: Float[Tensor, "*#batch"], y: Float[Tensor, "*#batch"], )

src/visualization/color_map.py:30

Functionapply_patch_shim

Crop images in the batch so that their dimensions are cleanly divisible by the specified patch size.

src/dataset/shims/patch_shim.py:30

Methodcalculate_auc

(self, tot_e_pose, method_name, overlap_tag)

src/evaluation/pose_evaluator.py:166

Methodcenter_principal_point

(image, cx, cy, h, w)

src/dataset/dataset_scannet_pose.py:139

Functioncollate

Merge nested dictionaries of tensors.

src/misc/collation.py:8

Functioncolmap_to_opencv_intrinsics

Modify camera intrinsics to follow a different convention. Coordinates of the center of the top-left pixels are by default: - (0.5, 0.5)

src/geometry/ptc_geometry.py:221

Functioncolor_function

( xy: Float[Tensor, "point 2"], )

src/visualization/drawing/points.py:40

Functioncombined_shim

src/dataset/data_module.py:27

Functioncompose_extrinsic_RT

Compose the standard form extrinsic matrix from RT. Batched I/O.

src/misc/cam_utils.py:16

Functioncompute_geodesic_distance_from_two_matrices

src/evaluation/metrics.py:55

Methodcompute_loss

(self, gt1, gt2, pred1, pred2, **kw)

src/loss/loss_point.py:167

Methodconfigure_optimizers

src/model/model_wrapper.py:679

Functionconvert_pixel_to_world

( xy: Float[Tensor, "*batch 2"], )

src/visualization/drawing/coordinate_conversion.py:39

Functionconvert_world_to_pixel

( xy: Float[Tensor, "*batch 2"], )

src/visualization/drawing/coordinate_conversion.py:34

src/model/encoder/common/gaussian_adapter.py:118

(self, dimensionality: int)

src/model/encodings/positional_encoding.py:35

src/model/encoder/backbone/backbone_resnet.py:99

src/model/encoder/backbone/backbone_croco.py:280

src/model/encoder/backbone/backbone_dino.py:78

src/model/encoder/backbone/backbone.py:28

src/model/encoder/backbone/backbone_croco_multiview.py:234

src/model/encoder/common/gaussian_adapter.py:114

Methoddata_shim

(batch: BatchedExample)

src/model/encoder/encoder_noposplat_multi.py:230

← previous next →301–400 of 553, ranked by callers