Functions — github.com/Livioni/OmniVGGT-official

Method__getitem__

(self, idx)

omnivggt/datasets/scannet.py:247

Method__getitem__

(self, idx)

omnivggt/datasets/mvs_synth.py:236

Method__getitem__

(self, idx)

omnivggt/datasets/vkitti.py:248

Method__getitem__

(self, idx)

omnivggt/datasets/tartanair.py:221

Method__getitem__

(self, idx)

omnivggt/datasets/uasol.py:234

Method__getitem__

(self, idx)

omnivggt/datasets/arkitscenes_high.py:217

Method__getitem__

(self, idx)

omnivggt/datasets/blendedmvs.py:200

Method__getitem__

(self, idx)

omnivggt/datasets/megadepth.py:232

Method__getitem__

(self, idx)

omnivggt/datasets/dynamic_replica.py:213

Method__getitem__

(self, idx)

omnivggt/datasets/kubric.py:247

Method__getitem__

(self, idx)

omnivggt/datasets/mp3d.py:234

Method__getitem__

(self, idx)

omnivggt/datasets/dl3dv.py:297

Method__getitem__

(self, idx)

omnivggt/datasets/hypersim.py:239

Method__getitem__

(self, idx)

omnivggt/datasets/scannetppv2.py:251

Method__getitem__

(self, idx)

omnivggt/datasets/waymo.py:235

Method__getitem__

(self, idx)

omnivggt/datasets/bedlam.py:198

Method__getitem__

(self, idx)

omnivggt/datasets/co3d.py:237

Method__getitem__

(self, idx)

omnivggt/datasets/mapfree.py:225

Method__getitem__

(self, idx)

omnivggt/datasets/spring.py:235

Method__getitem__

(self, idx)

omnivggt/datasets/unreal4k.py:243

Method__getitem__

(self, idx)

omnivggt/datasets/base/easy_dataset.py:59

Method__getitem__

(self, idx)

omnivggt/datasets/base/easy_dataset.py:104

Method__getitem__

(self, idx)

omnivggt/datasets/base/easy_dataset.py:144

Method__getitem__

(self, idx)

omnivggt/datasets/base/easy_dataset.py:172

Method__getitem__

(self, idx)

omnivggt/datasets/base/base_stereo_view_dataset.py:67

Method__getitem__

(self, idx)

omnivggt/heads/datasets/dl3dv.py:300

Method__init__

(self, camera=None, depth=None, point=None, track=None, **kwargs)

omnivggt/loss.py:27

Method__init__

(self)

omnivggt/viz.py:122

Method__init__

(self, name, device, fmt=":f")

omnivggt/utils/general.py:72

Method__init__

(self, num_batches, meters, real_meters, prefix="")

omnivggt/utils/general.py:101

Method__init__

(self, name: str, device: Optional[torch.device] = None, fmt: str = ":f")

omnivggt/utils/general.py:263

Method__init__

(self, pool_size, version='pt')

omnivggt/utils/po_utils/misc.py:40

Method__init__

(self, filename=None)

omnivggt/utils/po_utils/improc.py:430

Method__init__

(self, writer, global_step, log_freq=10, fps=8, scalar_freq=100, just_gif=False)

omnivggt/utils/po_utils/improc.py:463

Method__init__

Initializes the 2D RoPE module.

omnivggt/layers/rope.py:79

Method__init__

(self, drop_prob=None)

omnivggt/layers/drop_path.py:29

Method__init__

(self, hidden_dim: int = 512, out_dim: int = 1024)

omnivggt/layers/block.py:110

Method__init__

(self, input_dim: int = 1024, hidden_dim: int = 1024, out_dim: int = 1024)

omnivggt/layers/block.py:135

Method__init__

( self, dim: int, init_values: Union[float, Tensor] = 1e-5, inplace: bool = Fa

omnivggt/layers/layer_scale.py:16

Method__init__

( self, dim: int, num_heads: int = 8, qkv_bias: bool = True, proj_bias

omnivggt/layers/attention.py:22

Method__init__

( self, in_features: int, hidden_features: Optional[int] = None, out_features:

omnivggt/layers/mlp.py:17

Method__init__

( self, in_features: int, hidden_features: Optional[int] = None, out_features:

omnivggt/layers/swiglu_ffn.py:55

Method__init__

( self, img_size: Union[int, Tuple[int, int]] = 224, patch_size: Union[int, Tuple[int,

omnivggt/layers/patch_embed.py:37

Method__init__

Args: img_size (int, tuple): input image size patch_size (int, tuple): patch size in_chans (int): number

omnivggt/layers/vision_transformer.py:43

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/wildrgbd', use_cache = Fals

omnivggt/datasets/wildrgb.py:36

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/scannet', use_cache = False

omnivggt/datasets/scannet.py:32

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/mvs_synth', use_cache = Fal

omnivggt/datasets/mvs_synth.py:32

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/vkitti', use_cache = False,

omnivggt/datasets/vkitti.py:31

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/tartanair', use_cache = Fal

omnivggt/datasets/tartanair.py:35

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/uasol', use_cache = False,

omnivggt/datasets/uasol.py:32

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/arkitscene_highres', dset='

omnivggt/datasets/arkitscenes_high.py:29

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/blendedmvs', dset='',

omnivggt/datasets/blendedmvs.py:30

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/megadepth', use_cache = Fal

omnivggt/datasets/megadepth.py:38

Method__init__

(self, dataset_location='datasets/dynamic_replica', dset='train',

omnivggt/datasets/dynamic_replica.py:31

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/kubric', dset='trackings',

omnivggt/datasets/kubric.py:33

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/mp3d', use_cache = False,

omnivggt/datasets/mp3d.py:32

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/dl3dv', dset='',

omnivggt/datasets/dl3dv.py:33

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/hypersim', dset='',

omnivggt/datasets/hypersim.py:53

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/scannetppv2', use_cache = F

omnivggt/datasets/scannetppv2.py:32

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/waymo', use_cache = False,

omnivggt/datasets/waymo.py:32

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/bedlam', dset='',

omnivggt/datasets/bedlam.py:32

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/co3d', dset='',

omnivggt/datasets/co3d.py:33

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/mapfree', dset='',

omnivggt/datasets/mapfree.py:32

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/spring', use_cache = False,

omnivggt/datasets/spring.py:32

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/unreal4k', use_cache = Fals

omnivggt/datasets/unreal4k.py:35

Method__init__

(self, window_size=20, fmt=None)

omnivggt/datasets/utils/misc.py:35

Method__init__

(self, delimiter="\t")

omnivggt/datasets/utils/misc.py:93

Method__init__

(self, enabled=True)

omnivggt/datasets/utils/misc.py:252

Method__init__

(self, images)

omnivggt/datasets/utils/cropping.py:26

Method__init__

(self, new_size, dataset)

omnivggt/datasets/base/easy_dataset.py:76

Method__init__

(self, datasets)

omnivggt/datasets/base/easy_dataset.py:127

Method__init__

(self, datasets)

omnivggt/datasets/base/easy_dataset.py:168

Method__init__

(self, *, # only keyword arguments dset=None, resolution=None, # square_si

omnivggt/datasets/base/base_stereo_view_dataset.py:29

Method__init__

(self, dataset, batch_size, pool_size, world_size=1, rank=0, drop_last=True)

omnivggt/datasets/base/batched_sampler.py:22

Method__init__

(self, dataset, batch_size, pool_size, world_size=1, rank=0, drop_last=True, recent_buffer_si

omnivggt/datasets/base/batched_sampler.py:79

Method__init__

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

omnivggt/heads/dpt_head.py:360

Method__init__

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

omnivggt/heads/dpt_head.py:405

Method__init__

( self, dim_in: int = 2048, trunk_depth: int = 4, pose_encoding_type: str = "a

omnivggt/heads/camera_head.py:26

Method__init__

(self, dataset_location='/mnt/disk3.8-4/datasets/dl3dv', dset='',

omnivggt/heads/datasets/dl3dv.py:32

Method__init__

( self, img_size=518, patch_size=14, embed_dim=1024, depth=24,

omnivggt/models/aggregator.py:52

Method__init__

(self, img_size=518, patch_size=14, embed_dim=1024, dept

omnivggt/models/omnivggt_aggregator.py:20

Method__init__

(self, img_size=518, patch_size=14, embed_dim=1024, cam_drop_prob=0.1, depth_drop_prob=0.1, e

omnivggt/models/omnivggt.py:11

Method__iter__

(self)

omnivggt/datasets/base/batched_sampler.py:41

Method__iter__

(self)

omnivggt/datasets/base/batched_sampler.py:90

Method__len__

(self)

omnivggt/utils/po_utils/misc.py:49

Method__len__

(self)

omnivggt/datasets/wildrgb.py:182

Method__len__

(self)

omnivggt/datasets/scannet.py:177

Method__len__

(self)

omnivggt/datasets/mvs_synth.py:175

Method__len__

(self)

omnivggt/datasets/vkitti.py:181

Method__len__

(self)

omnivggt/datasets/tartanair.py:164

Method__len__

(self)

omnivggt/datasets/uasol.py:175

Method__len__

(self)

omnivggt/datasets/arkitscenes_high.py:160

Method__len__

(self)

omnivggt/datasets/blendedmvs.py:151

Method__len__

(self)

omnivggt/datasets/megadepth.py:174

Method__len__

(self)

omnivggt/datasets/dynamic_replica.py:161

Method__len__

(self)

omnivggt/datasets/kubric.py:172

Method__len__

(self)

omnivggt/datasets/mp3d.py:174

Method__len__

(self)

omnivggt/datasets/dl3dv.py:176

Method__len__

(self)

omnivggt/datasets/hypersim.py:184

Method__len__

(self)

omnivggt/datasets/scannetppv2.py:184

Functions660 in github.com/Livioni/OmniVGGT-official