MCPcopy Create free account

hub / github.com/HavenFeng/St4RTrack / functions

Functions1,146 in github.com/HavenFeng/St4RTrack

⨍ Functions 1,146◇ Types & classes 143
Method__call__
(self, X)
dust3r/utils/po_utils/improc.py:437
Method__call__
(self, loss, optimizer, clip_grad=None, parameters=None, create_graph=False, update_grad=True)
croco/utils/misc.py:258
Method__call__
(self, img1, img2, disp, dataset_name)
croco/stereoflow/augmentor.py:151
Method__call__
(self, img1, img2, flow, dname)
croco/stereoflow/augmentor.py:284
Method__call__
(self, img1, img2)
croco/datasets/transforms.py:24
Method__call__
(self, x)
croco/models/masking.py:22
Method__call__
(self, b, h, w, device)
croco/models/blocks.py:435
Method__call__
(self, b, t, h, w, device)
croco/models/blocks.py:449
Method__del__
(self)
croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:221
Method__getattr__
(self, attr)
croco/utils/misc.py:109
Method__getitem__
(self, index)
dust3r/datasets/DR_traj_loader.py:192
Method__getitem__
(self, idx)
dust3r/datasets/base/easy_dataset.py:143
Method__getitem__
(self, idx)
dust3r/datasets/base/easy_dataset.py:183
Method__getitem__
(self, idx)
dust3r/datasets/base/easy_dataset.py:222
Method__getitem__
(self, idx)
dust3r/datasets/base/base_stereo_view_dataset.py:69
Method__getitem__
(self, index)
croco/stereoflow/datasets_flow.py:60
Method__getitem__
(self, index)
croco/stereoflow/datasets_stereo.py:75
Method__getitem__
(self, index)
croco/datasets/pairs_dataset.py:95
Method__getitem__
(self, useless_idx)
croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:282
Method__init__
(self, pred_intrinsics=False, criterion=None)
dust3r/camera_solver.py:13
Method__init__
(self, output_mode='pts3d', head_type='linear', depth_mode=
dust3r/model_seq.py:56
Method__init__
(self, reduction='mean')
dust3r/losses.py:195
Method__init__
(self, criterion=None)
dust3r/losses.py:279
Method__init__
(self)
dust3r/losses.py:303
Method__init__
(self, pixel_loss, alpha=1, velo_weight=1, pose_weight=0, traj_weight=0, align3d_weight=0, de
dust3r/losses.py:443
Method__init__
( self, patch_size=16, in_channels=3, embed_dim=768, bias=True, )
dust3r/rope3d.py:12
Method__init__
(self, )
dust3r/rope3d.py:36
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768, norm_layer=None, flatten=True, init='xavier')
dust3r/patch_embed.py:40
Method__init__
(self, output_mode='pts3d', head_type1='dpt', head1_pretrai
dust3r/model.py:60
Method__init__
(self)
dust3r/viz.py:122
Method__init__
(self)
dust3r/utils/goem_opt.py:39
Method__init__
(self, init_focal_length=0.45, pixel_size=None)
dust3r/utils/goem_opt.py:73
Method__init__
(self, number_of_points=10)
dust3r/utils/goem_opt.py:309
Method__init__
(self, n_points=10, use_inverse=False, grid_size=1)
dust3r/utils/goem_opt.py:375
Method__init__
(self)
dust3r/utils/goem_opt.py:431
Method__init__
(self)
dust3r/utils/goem_opt.py:531
Method__init__
( self, save_dir: str = "./results", grayscale: bool = False, pad_value: int =
dust3r/utils/cotracker_vis.py:63
Method__init__
(self, pool_size, version='pt')
dust3r/utils/po_utils/misc.py:40
Method__init__
(self, filename=None)
dust3r/utils/po_utils/improc.py:430
Method__init__
(self, writer, global_step, log_freq=10, fps=8, scalar_freq=100, just_gif=False)
dust3r/utils/po_utils/improc.py:463
Method__init__
(self, *args, ROOT, **kwargs)
dust3r/datasets/scannetpp.py:19
Method__init__
(self, dataset_location='data/tartanair', dset='Hard', use_
dust3r/datasets/tartanair.py:35
Method__init__
(self, ROOT, *args, mask_bg='rand', **kwargs)
dust3r/datasets/staticthings3d.py:19
Method__init__
(self, # only keyword arguments dataset_location = "./Davis_data/DAVIS/JPEGImages/480p/roller
dust3r/datasets/custom.py:104
Method__init__
(self, size, *args, ROOT, **kwargs)
dust3r/datasets/habitat.py:21
Method__init__
(self, *args, ROOT, split=None, **kwargs)
dust3r/datasets/blendedmvs.py:20
Method__init__
(self, dataset_location='data/sintel/training', dset='clean',
dust3r/datasets/sintel.py:50
Method__init__
(self, dataset_location='/datasets/kubric_full_frame_fp32_cam_pos_batch', ds
dust3r/datasets/kubrick.py:166
Method__init__
(self, *args, split, ROOT, **kwargs)
dust3r/datasets/megadepth.py:17
Method__init__
(self, dataset_location='/dynamic_stereo/dynamic_replica_data/train', use_au
dust3r/datasets/dynamic_replica.py:138
Method__init__
(self, dataset_location='data/pointodyssey', dset='train',
dust3r/datasets/pointodyssey.py:54
Method__init__
(self, *args, split, ROOT, **kwargs)
dust3r/datasets/arkitscenes.py:18
Method__init__
(self, # only keyword arguments dataset_location = ["/is/cluster/fast/groups/ps-pbrgaussians/
dust3r/datasets/megasam_delta.py:38
Method__init__
(self, dataset_location='data/spring', dset='train', use_au
dust3r/datasets/spring_dataset.py:33
Method__init__
(self, *args, ROOT, pairs_npz_name='waymo_pairs_video.npz', **kwargs)
dust3r/datasets/waymo.py:23
Method__init__
( self, root='/fsx-repligen/nikitakaraev/datasets/dynamic_replica', split="valid",
dust3r/datasets/DR_traj_loader.py:106
Method__init__
(self, mask_bg=True, *args, ROOT, **kwargs)
dust3r/datasets/co3d.py:24
Method__init__
(self, mask_bg=True, *args, ROOT, **kwargs)
dust3r/datasets/wildrgbd.py:19
Method__init__
(self, images)
dust3r/datasets/utils/cropping.py:25
Method__init__
(self, *args, **kwargs)
dust3r/datasets/base/easy_dataset.py:21
Method__init__
(self, new_size, dataset, curriculum_learning=False)
dust3r/datasets/base/easy_dataset.py:50
Method__init__
(self, datasets)
dust3r/datasets/base/easy_dataset.py:160
Method__init__
(self, multiplicator, dataset)
dust3r/datasets/base/easy_dataset.py:211
Method__init__
(self, *, split=None, resolution=None, # square_size or (width, height) o
dust3r/datasets/base/base_stereo_view_dataset.py:29
Method__init__
(self, dataset, batch_size, pool_size, world_size=1, rank=0, drop_last=True)
dust3r/datasets/base/batched_sampler.py:21
Method__init__
(self, *, n_cls_token=0, hooks_idx=None, dim_tokens=None, output_width_ratio=1, num_channels=
dust3r/heads/dpt_head.py:71
Method__init__
(self, net, has_conf=False)
dust3r/heads/linear_head.py:18
Method__init__
( self, num_points=16, num_proposals=64, num_iter=3,
dust3r/cloud_opt/levenberg_marquardt.py:273
Method__init__
(self, *args, acg_mle_iter=3, acg_dispersion=0.001,
dust3r/cloud_opt/epropnp.py:198
Method__init__
(self, *args, **kwargs)
dust3r/cloud_opt/pair_viewer.py:24
Method__init__
(self, pts3d_i, # (B, H, W, 3) pts3d_j, # (B, H, W, 3)
dust3r/cloud_opt/oneref_viewer.py:127
Method__init__
Args: cam_mats (Tensor): Shape (*, 3, 3) img_shape (Tensor | None): Shape (*, 2) in [h, w] lb (Tensor | N
dust3r/cloud_opt/camera.py:35
Method__init__
(self, scale_tril, validate_args=None, eps=1e-6)
dust3r/cloud_opt/distributions.py:19
Method__init__
(self, delta=None, relative_delta=0.5, eps=1e-10)
dust3r/cloud_opt/cost_fun.py:114
Method__init__
(self, window_size=20, fmt=None)
croco/utils/misc.py:32
Method__init__
(self, delimiter="\t")
croco/utils/misc.py:96
Method__init__
(self, enabled=True)
croco/utils/misc.py:255
Method__init__
(self, crop_size, scale_prob=0.5, scale_xonly=True, lhth=800., lminscale=0.0, lmaxscale=1.0, hminscale=-0.2, h
croco/stereoflow/augmentor.py:27
Method__init__
(self, crop_size, min_scale=-0.2, max_scale=0.5, spatial_aug_prob=0.8, stretch_prob=0.8, max_stretch=0.2, h_fl
croco/stereoflow/augmentor.py:163
Method__init__
(self, split, augmentor=False, crop_size=None, totensor=True)
croco/stereoflow/datasets_flow.py:38
Method__init__
(self, max_gtnorm=None)
croco/stereoflow/criterion.py:43
Method__init__
(self, max_gtnorm=10000., a=0.25, b=4.)
croco/stereoflow/criterion.py:57
Method__init__
(self, max_gtnorm=None, a=3.0, b=3.0)
croco/stereoflow/criterion.py:72
Method__init__
(self, do_quantile=False)
croco/stereoflow/criterion.py:90
Method__init__
(self)
croco/stereoflow/criterion.py:114
Method__init__
(self)
croco/stereoflow/criterion.py:142
Method__init__
(self)
croco/stereoflow/criterion.py:186
Method__init__
(self, split, augmentor=False, crop_size=None, totensor=True)
croco/stereoflow/datasets_stereo.py:53
Method__init__
(self, assymetric_prob, **kwargs)
croco/datasets/transforms.py:38
Method__init__
(self, dnames, trfs='', totensor=True, normalize=True, data_dir='./data/')
croco/datasets/pairs_dataset.py:87
Method__init__
(self, *args)
croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:118
Method__init__
(self, scene, navmesh, scene_dataset_config_file,
croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:122
Method__init__
(self, num_patches, mask_ratio)
croco/models/masking.py:17
Method__init__
(self, drop_prob: float = 0., scale_by_keep: bool = True)
croco/models/blocks.py:48
Method__init__
(self, dim, rope=None, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0., zero_init=False)
croco/models/blocks.py:86
Method__init__
( self, dim, num_heads=8, qkv_bias=False, attn_drop=0.0, proj_drop=0.0, r=4, alpha=1.0, rope=N
croco/models/blocks.py:148
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, drop=0., attn_drop=0., drop_path=0., act
croco/models/blocks.py:221
Method__init__
(self, dim, rope=None, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0.)
croco/models/blocks.py:239
Method__init__
( self, dim, num_heads=8, qkv_bias=False, attn_drop=0.0, proj_drop=0.0, r=4, alpha=1.0, rope=N
croco/models/blocks.py:292
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, drop=0., attn_drop=0., drop_path=0., act
croco/models/blocks.py:370
← previousnext →601–700 of 1,146, ranked by callers