Functions — github.com/Lijiaxin0111/Open-d4rt

hub / github.com/Lijiaxin0111/Open-d4rt / functions

Functions626 in github.com/Lijiaxin0111/Open-d4rt

⨍ Functions 626◇ Types & classes 62

↓ 1 callersFunctionmain

()

vis/serve_demo_viser.py:223

↓ 1 callersFunctionmain

()

src/data_preprocess/prepare_kubric_full_preprocessed.py:435

↓ 1 callersFunctionmain

()

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:674

↓ 1 callersFunctionmasked_mean_per_sample

(values: torch.Tensor, mask: torch.Tensor, eps: float = 1e-6)

src/model/utils.py:37

↓ 1 callersMethodmaybe_save_best

(self, metrics: dict[str, float])

src/engine/trainer.py:512

↓ 1 callersMethodmaybe_save_step_checkpoint

(self)

src/engine/trainer.py:498

↓ 1 callersFunctionparse_args

()

eval_track3d_in_worldtrack.py:107

↓ 1 callersFunctionparse_args

()

train.py:66

↓ 1 callersFunctionparse_args

()

vis/build_like_demo_for_worldtrack.py:41

↓ 1 callersFunctionparse_args

()

vis/serve_demo_viser.py:19

↓ 1 callersFunctionparse_args

()

src/data_preprocess/prepare_kubric_full_preprocessed.py:418

↓ 1 callersFunctionpreprocess_split

( *, input_root: Path, output_root: Path, split: str, max_scenes: int | None, overwrit

src/data_preprocess/prepare_kubric_full_preprocessed.py:379

↓ 1 callersFunctionrender_track_comparison_videos

( *, video_rgb: np.ndarray, gt_tracks_world: np.ndarray, pred_tracks_world: np.ndarray, vi

vis/build_like_demo_for_worldtrack.py:271

↓ 1 callersMethodresume

(self, ckpt_path: str | None)

src/engine/trainer.py:534

↓ 1 callersFunctionsave_sample

Save one TFDS numpy sample and return its final scene directory.

src/data_preprocess/prepare_kubric_full_preprocessed.py:199

↓ 1 callersMethodset_dataset_epoch

(self, epoch: int)

src/data/seeding.py:68

↓ 1 callersFunctionset_dataset_epoch_recursive

(dataset: Any, epoch: int)

src/data/seeding.py:110

↓ 1 callersMethodset_dataset_worker

(self, worker_id: int)

src/data/seeding.py:71

↓ 1 callersFunctionset_dataset_worker_recursive

(dataset: Any, worker_id: int)

src/data/seeding.py:117

↓ 1 callersMethodshould_log_train

(self, step: int)

src/vis/training_visualizer.py:360

↓ 1 callersMethodshould_log_val

(self, step: int)

src/vis/training_visualizer.py:369

↓ 1 callersFunctionsinusoidal_position_embedding

(length: int, dim: int, device: torch.device)

src/model/utils.py:20

↓ 1 callersFunctionwrite_manifest

( output_root: Path, scenes: list[SavedScene], *, input_root: Path, split_options: dict[st

src/data_preprocess/prepare_kubric_full_preprocessed.py:341

Function_

(_event: Any)

vis/serve_demo_viser.py:480

Method__getattr__

(self, item: str)

src/core/config.py:31

Method__getitem__

(self, index: int)

src/data/scannet_raw_dataset.py:870

Method__getitem__

(self, index: int)

src/data/dynamic_replica_raw_dataset.py:760

Method__getitem__

(self, index: int)

src/data/kubric_full_robust_dataset.py:1177

Method__getitem__

(self, index: int)

src/data/pointodyssey_raw_dataset.py:401

Method__getitem__

(self, index: int)

src/data/blendermvs_raw_dataset.py:491

Method__getitem__

(self, index: int)

src/data/mixture_dataset.py:65

Method__getitem__

(self, index: int)

src/data/virtual_kitti2_raw_dataset.py:838

Method__getitem__

(self, index: int)

src/data/tartanair_raw_dataset.py:406

Method__getitem__

(self, index: int)

src/data/kubric_full_robust_preprocess_dataset.py:164

Method__getitem__

(self, index: int)

src/data/mvs_synth_raw_dataset.py:416

Method__getitem__

(self, index: int)

src/data/co3d_raw_dataset.py:558

Method__init__

(self, source: dict[str, Any] | None = None)

src/core/config.py:25

Method__init__

(self, output_path: str | Path)

src/core/logging.py:35

Method__init__

(self, name: str)

src/core/registry.py:11

Method__init__

(self, hidden_dim: int, num_layers: int, num_heads: int, mlp_ratio: float)

src/model/decoder.py:46

Method__init__

(self, model_cfg: Any)

src/model/d4rt.py:172

Method__init__

(self, input_dim: int = 2, num_bands: int = 8, include_input: bool = True)

src/model/query_embedding.py:13

Method__init__

(self, hidden_dim: int)

src/model/heads.py:12

Method__init__

( self, in_channels: int, hidden_dim: int, patch_size_t_h_w: tuple[int, int, i

src/model/encoder.py:60

Method__init__

(self, config: ScannetRawConfig)

src/data/scannet_raw_dataset.py:421

Method__init__

(self, config: DynamicReplicaRawConfig)

src/data/dynamic_replica_raw_dataset.py:215

Method__init__

(self, config: KubricFullRobustConfig)

src/data/kubric_full_robust_dataset.py:190

Method__init__

(self, config: PointOdysseyRawConfig)

src/data/pointodyssey_raw_dataset.py:112

Method__init__

(self, config: BlendermvsRawConfig)

src/data/blendermvs_raw_dataset.py:195

Method__init__

(self, config: MixtureDatasetConfig)

src/data/mixture_dataset.py:42

Method__init__

(self, config: VirtualKitti2RawConfig)

src/data/virtual_kitti2_raw_dataset.py:439

Method__init__

(self, message: str, failed_paths: list[str] | None = None)

src/data/bad_sample_registry.py:20

Method__init__

(self, config: TartanairRawConfig)

src/data/tartanair_raw_dataset.py:156

Method__init__

(self, config: KubricFullRobustPreprocessConfig)

src/data/kubric_full_robust_preprocess_dataset.py:62

Method__init__

(self, config: MvsSynthRawConfig)

src/data/mvs_synth_raw_dataset.py:164

Method__init__

(self, config: Co3dRawConfig)

src/data/co3d_raw_dataset.py:172

Method__init__

(self, optimizer: torch.optim.Optimizer, warmup_steps: int, total_steps: int, peak_lr: float, final_lr: float)

src/engine/trainer.py:137

Method__init__

( self, model: torch.nn.Module, loss_fn: torch.nn.Module, train_loader: Any,

src/engine/trainer.py:189

Method__init__

(self, train_cfg: Any, output_dir: Path, tb_writer: Any | None, logger: Any | None = None)

src/vis/training_visualizer.py:327

Method__init__

(self, train_cfg: Any)

src/losses/d4rt_loss.py:14

Method__len__

(self)

src/data/scannet_raw_dataset.py:448

Method__len__

(self)

src/data/dynamic_replica_raw_dataset.py:533

Method__len__

(self)

src/data/kubric_full_robust_dataset.py:409

Method__len__

(self)

src/data/pointodyssey_raw_dataset.py:264

Method__len__

(self)

src/data/blendermvs_raw_dataset.py:354

Method__len__

(self)

src/data/mixture_dataset.py:62

Method__len__

(self)

src/data/virtual_kitti2_raw_dataset.py:562

Method__len__

(self)

src/data/tartanair_raw_dataset.py:285

Method__len__

(self)

src/data/mvs_synth_raw_dataset.py:284

Method__len__

(self)

src/data/co3d_raw_dataset.py:424

Method__setattr__

(self, key: str, value: Any)

src/core/config.py:36

Function_align_pred_to_gt_scale_only

( *, pred_xyz_ref0: np.ndarray, gt_xyz_ref0: np.ndarray, valid_mask: np.ndarray, )

vis/build_like_demo.py:695

Function_amp_dtype_for_device

(device: torch.device)

src/engine/trainer.py:35

Function_apply_vertical_crop_to_camera_data

(camera_data: dict[str, np.ndarray], top: int)

vis/build_like_demo.py:356

Function_build_blendedmvs_raw

(split: str, cfg: Any, manifest_paths: list[str] | None = None)

src/data/builder.py:210

Function_build_co3d_raw

(split: str, cfg: Any, manifest_paths: list[str] | None = None)

src/data/builder.py:253

Function_build_d4rt

(model_cfg: Any)

src/model/builder.py:15

Function_build_d4rt_loss

(train_cfg: Any)

src/losses/builder.py:15

Function_build_dynamic_replica_raw

(split: str, cfg: Any, manifest_paths: list[str] | None = None)

src/data/builder.py:416

Function_build_kubric_full_robust

(split: str, cfg: Any, manifest_paths: list[str] | None = None)

src/data/builder.py:284

Function_build_kubric_full_robust_preprocess

(split: str, cfg: Any, manifest_paths: list[str] | None = None)

src/data/builder.py:315

Function_build_mixture_raw

(split: str, cfg: Any, manifest_paths: list[str] | None = None)

src/data/builder.py:643

Function_build_mvs_synth_raw

(split: str, cfg: Any, manifest_paths: list[str] | None = None)

src/data/builder.py:451

Function_build_pointodyssey_raw

(split: str, cfg: Any, manifest_paths: list[str] | None = None)

src/data/builder.py:347

Function_build_scannet_raw

(split: str, cfg: Any, manifest_paths: list[str] | None = None)

src/data/builder.py:500

Function_build_tartanair_raw

(split: str, cfg: Any, manifest_paths: list[str] | None = None)

src/data/builder.py:525

Function_build_virtual_kitti2_raw

(split: str, cfg: Any, manifest_paths: list[str] | None = None)

src/data/builder.py:376

Method_clip_start

(self, scene: _Scene, index: int)

src/data/dynamic_replica_raw_dataset.py:543

Function_crop_video_vertical

(video_rgb: np.ndarray, top: int, bottom: int)

vis/build_like_demo.py:97

Function_detect_vertical_valid_crop

(video_rgb: np.ndarray, black_bar_threshold: float)

vis/build_like_demo.py:79

Method_frame_idx

(p: Path)

src/data/tartanair_raw_dataset.py:225

Function_load_kubric_dynamic_masks

( *, tfds_root: Path, sample_json_path: Path, expected_num_frames: int, crop_top: int,

vis/build_like_demo.py:721

Function_load_kubric_gt_tracks

( *, tfds_root: Path, sample_json_path: Path, track_query_uv_px: np.ndarray, track_query_t

vis/build_like_demo.py:545

Function_load_video_rgb

(path: Path, max_frames: int)

infer_track_3d.py:46

Function_on_click

(event: Any)

vis/serve_demo_viser.py:460

Method_project_points

(frame_idx: int, points_world: np.ndarray, k_use: np.ndarray)

src/data/virtual_kitti2_raw_dataset.py:665

Method_raise_or_warn

(message: str)

src/model/d4rt.py:295

Method_reset_eval_iter

(self)

src/data/kubric_full_robust_preprocess_dataset.py:90

Method_reset_train_iter

(self)

src/data/kubric_full_robust_preprocess_dataset.py:87

Function_safe_name

(value: str)

src/data_preprocess/prepare_kubric_full_preprocessed.py:125

← previous next →501–600 of 626, ranked by callers