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 callersMethod_sample_key

(self, scene: _Scene, idxs: list[int])

src/data/blendermvs_raw_dataset.py:370

↓ 1 callersMethod_sample_key

(self, scene: _Scene, idxs: list[int])

src/data/virtual_kitti2_raw_dataset.py:578

↓ 1 callersMethod_sample_key

(self, traj: _Trajectory, idxs: list[int])

src/data/tartanair_raw_dataset.py:300

↓ 1 callersMethod_sample_key

(self, scene: _Scene, idxs: list[int])

src/data/mvs_synth_raw_dataset.py:300

↓ 1 callersMethod_sample_key

(self, scene: _Scene, idxs: list[int])

src/data/co3d_raw_dataset.py:440

↓ 1 callersMethod_sample_key_dslr

(self, scene: _DSLRScene, idxs: list[int])

src/data/scannet_raw_dataset.py:860

↓ 1 callersMethod_sample_key_iphone

(self, scene: _IphoneScene, idxs: list[int])

src/data/scannet_raw_dataset.py:852

↓ 1 callersMethod_sample_paths

(self, scene: _Scene, frame_ids: list[int])

src/data/dynamic_replica_raw_dataset.py:590

↓ 1 callersMethod_sample_paths

(self, scene: _Scene, idxs: list[int])

src/data/pointodyssey_raw_dataset.py:293

↓ 1 callersMethod_sample_paths

(self, scene: _Scene, idxs: list[int])

src/data/blendermvs_raw_dataset.py:374

↓ 1 callersMethod_sample_paths

(self, scene: _Scene, idxs: list[int])

src/data/virtual_kitti2_raw_dataset.py:582

↓ 1 callersMethod_sample_paths

(self, traj: _Trajectory, idxs: list[int])

src/data/tartanair_raw_dataset.py:304

↓ 1 callersMethod_sample_paths

(self, scene: _Scene, idxs: list[int])

src/data/mvs_synth_raw_dataset.py:304

↓ 1 callersMethod_sample_paths

(self, scene: _Scene, idxs: list[int])

src/data/co3d_raw_dataset.py:444

↓ 1 callersMethod_sample_paths_dslr

(self, scene: _DSLRScene, idxs: list[int])

src/data/scannet_raw_dataset.py:864

↓ 1 callersMethod_sample_paths_iphone

(self, scene: _IphoneScene)

src/data/scannet_raw_dataset.py:856

↓ 1 callersFunction_sample_rgb_from_uv_sequence

(video_rgb: np.ndarray, uv_px: np.ndarray)

vis/build_like_demo.py:61

↓ 1 callersFunction_sample_scale

()

src/data/raw_augment.py:511

↓ 1 callersFunction_sample_t_tgt_from_source

Sample target timesteps from a local/global mixture around source time.

src/data/raw_augment.py:338

↓ 1 callersFunction_sample_track_subset

( gt_tracks_world: np.ndarray, pred_tracks_world: np.ndarray, visibility_tq: np.ndarray, max_p

vis/build_like_demo_for_worldtrack.py:224

↓ 1 callersFunction_sample_valid_target_time

(fs: int, valid_tgt: np.ndarray)

src/data/virtual_kitti2_raw_dataset.py:211

↓ 1 callersFunction_save_array

(path: Path, array: Any)

src/data_preprocess/prepare_kubric_full_preprocessed.py:176

↓ 1 callersFunction_scale_per_trajectory

(gt_points: np.ndarray, pred_points: np.ndarray)

eval_track3d_in_worldtrack.py:147

↓ 1 callersFunction_scannet_split_file

(split: str, cfg: Any)

src/data/builder.py:485

↓ 1 callersMethod_scene

(self, index: int)

src/data/scannet_raw_dataset.py:477

↓ 1 callersMethod_scene

(self, index: int)

src/data/pointodyssey_raw_dataset.py:271

↓ 1 callersMethod_scene

(self, index: int)

src/data/blendermvs_raw_dataset.py:358

↓ 1 callersMethod_scene

(self, index: int)

src/data/virtual_kitti2_raw_dataset.py:566

↓ 1 callersMethod_scene

(self, index: int)

src/data/mvs_synth_raw_dataset.py:288

↓ 1 callersMethod_scene

(self, index: int)

src/data/co3d_raw_dataset.py:428

↓ 1 callersMethod_scene_for_index

(self, index: int)

src/data/dynamic_replica_raw_dataset.py:537

↓ 1 callersFunction_scene_name

(sample: dict[str, Any], index: int)

src/data_preprocess/prepare_kubric_full_preprocessed.py:131

↓ 1 callersMethod_scene_size

(self, scene_name: str, scene_dir: Path)

src/data/dynamic_replica_raw_dataset.py:522

↓ 1 callersFunction_seg_boundary_mask

(seg_hw: np.ndarray)

src/data/kubric_full_robust_dataset.py:53

↓ 1 callersFunction_select_dynamic_interior_track_queries

( *, query_uv_px: np.ndarray, dynamic_mask_hw: np.ndarray, max_tracks: int, allowed_mask:

vis/build_like_demo.py:951

↓ 1 callersFunction_select_query_indices

(t_tgt: np.ndarray, uv_mask: np.ndarray)

src/vis/training_visualizer.py:87

↓ 1 callersFunction_select_worldtrack_track_indices

( *, gt_tracks_world_tq3: np.ndarray, visibility_tq: np.ndarray, max_tracks: int, min_visi

vis/build_like_demo_for_worldtrack.py:188

↓ 1 callersMethod_selected_scene_names

(self)

src/data/virtual_kitti2_raw_dataset.py:542

↓ 1 callersFunction_set_3d_equal_axes

(ax: Any, points_xyz: np.ndarray)

src/vis/training_visualizer.py:48

↓ 1 callersFunction_sha256_file

(path: Path)

train.py:108

↓ 1 callersFunction_slice_batch

(value: Any, limit: int)

src/engine/trainer.py:83

↓ 1 callersFunction_solve_scale_only

(pred: np.ndarray, gt: np.ndarray)

src/vis/training_visualizer.py:110

↓ 1 callersFunction_solve_scale_only

(pred: np.ndarray, gt: np.ndarray)

src/eval/tasks.py:38

↓ 1 callersFunction_solve_scale_shift

(pred: np.ndarray, gt: np.ndarray)

src/vis/training_visualizer.py:117

↓ 1 callersFunction_source_record_plan

( *, tfds_dir: Path, split: str, shard_start: int | None = None, shard_end: int | None = N

src/data_preprocess/prepare_kubric_full_preprocessed.py:257

↓ 1 callersMethod_split_dir_name

(self)

src/data/pointodyssey_raw_dataset.py:160

↓ 1 callersFunction_split_manifest_options

( scenes: list[SavedScene], *, max_scenes_per_split: int | None, shard_start: int | None,

src/data_preprocess/prepare_kubric_full_preprocessed.py:321

↓ 1 callersFunction_split_overrides

(overrides: list[str] | None)

train.py:49

↓ 1 callersFunction_split_shard_lengths

(tfds_dir: Path, split: str)

src/data_preprocess/prepare_kubric_full_preprocessed.py:249

↓ 1 callersFunction_stable_namespace_seed

(namespace: str)

src/data/seeding.py:29

↓ 1 callersMethod_start

(self, scene: _IphoneScene | _DSLRScene, index: int)

src/data/scannet_raw_dataset.py:489

↓ 1 callersFunction_step_from_ckpt

(path: Path)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:180

↓ 1 callersFunction_structured_pretrained_tensor

(dst_key: str, dst_value: torch.Tensor, src_state: dict[str, torch.Tensor])

src/model/d4rt.py:107

↓ 1 callersFunction_subsample_indices

(total: int, keep: int)

vis/serve_demo_viser.py:154

↓ 1 callersFunction_subsample_indices

(mask: np.ndarray, max_points: int)

src/vis/training_visualizer.py:102

↓ 1 callersFunction_summary_num_frames

(payload: dict[str, Any])

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:555

↓ 1 callersMethod_sync_non_finite_status

(self, bad_key: str | None)

src/engine/trainer.py:642

↓ 1 callersMethod_tb_log_per_dataset_metrics

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

src/engine/trainer.py:886

↓ 1 callersMethod_tb_log_train_payload

(self, payload: dict[str, Any])

src/engine/trainer.py:691

↓ 1 callersFunction_to_int_list

(value: Any)

src/data/builder.py:172

↓ 1 callersFunction_to_split_bucket

(name: str, modulo: int = 20)

src/data/blendermvs_raw_dataset.py:37

↓ 1 callersFunction_to_split_bucket

(name: str, modulo: int = 20)

src/data/mvs_synth_raw_dataset.py:39

↓ 1 callersMethod_token_cap

(self, x: torch.Tensor)

src/model/encoder.py:101

↓ 1 callersFunction_track_colors

(n: int)

vis/build_like_demo_for_worldtrack.py:214

↓ 1 callersMethod_trajectory

(self, index: int)

src/data/tartanair_raw_dataset.py:289

↓ 1 callersFunction_transform_world_to_ref

(points_world_tn3: np.ndarray, t_wc_ref: np.ndarray)

vis/build_like_demo.py:536

↓ 1 callersMethod_try_add_trajectory

(self, scene_name: str, difficulty: str, traj_dir: Path, cam: str)

src/data/tartanair_raw_dataset.py:201

↓ 1 callersFunction_umeyama_rigid

(src: np.ndarray, dst: np.ndarray)

src/eval/tasks.py:50

↓ 1 callersFunction_umeyama_sim3

(src: np.ndarray, dst: np.ndarray)

src/eval/tasks.py:11

↓ 1 callersFunction_unpack_state_dict

(payload: Any)

src/model/d4rt.py:30

↓ 1 callersFunction_unproject_background_world

( *, u: int, v: int, depth_range_value: float, k: np.ndarray, t_wc: np.ndarray, )

vis/build_like_demo.py:513

↓ 1 callersFunction_unwrap_state_dict

(payload: Any)

train.py:157

↓ 1 callersMethod_validate_config

(self)

src/data/dynamic_replica_raw_dataset.py:363

↓ 1 callersMethod_validate_per_dataset

(self)

src/engine/trainer.py:837

↓ 1 callersFunction_validate_sample

(sample: dict[str, Any])

src/data_preprocess/prepare_kubric_full_preprocessed.py:137

↓ 1 callersFunction_viewpoint_to_camera

( vp: dict[str, Any], image_h: int, image_w: int, camera_convention: str, )

src/data/dynamic_replica_raw_dataset.py:107

↓ 1 callersFunction_viewpoint_to_camera

Convert CO3D viewpoint to OpenCV-convention K, T_cw, T_wc. CO3D stores R, T in PyTorch3D right-multiply convention: X_cam_pt3d = X_world

src/data/co3d_raw_dataset.py:68

↓ 1 callersMethod_wanted_frames_for_split

(self, category_dir: Path)

src/data/co3d_raw_dataset.py:219

↓ 1 callersMethod_warn_depth_fallback

(self, scene: _Scene, reason: str)

src/data/dynamic_replica_raw_dataset.py:260

↓ 1 callersFunction_with_image_opacity

(image_rgb_u8: np.ndarray, opacity: float)

vis/serve_demo_viser.py:29

↓ 1 callersFunction_worker_count

(split: str, cfg: Any)

src/data/builder.py:697

↓ 1 callersFunction_world_to_cam

(x_world: np.ndarray, t_cw: np.ndarray)

src/data/scannet_raw_dataset.py:96

↓ 1 callersFunction_worldtrack_gt_camera_data

(sample: dict[str, Any])

vis/build_like_demo_for_worldtrack.py:165

↓ 1 callersFunction_wrapped_keys

(base_key: str)

src/model/d4rt.py:67

↓ 1 callersFunctionalign_tracks_global

(gt_tracks_world: np.ndarray, pred_tracks_ref0: np.ndarray)

vis/build_like_demo_for_worldtrack.py:183

↓ 1 callersFunctionbuild_dataset

(split: str, cfg: Any, manifest_arg: str | None = None)

src/data/builder.py:677

↓ 1 callersFunctionbuild_loss

(model_name: str, train_cfg: Any)

src/losses/builder.py:19

↓ 1 callersFunctionbuild_worldtrack_demo_package

( *, model: torch.nn.Module, cfg: Any, npz_path: Path, output_dir: Path, num_frames: i

vis/build_like_demo_for_worldtrack.py:473

↓ 1 callersFunctionclear_scene

()

vis/serve_demo_viser.py:311

↓ 1 callersMethodconfigure_dataset_seed

(self, base_seed: int)

src/data/seeding.py:63

↓ 1 callersFunctionconfigure_dataset_seeding

(dataset: Any, base_seed: int)

src/data/seeding.py:103

↓ 1 callersMethoddecode_queries

( self, video: torch.Tensor, query: dict[str, torch.Tensor], memory: torch.Ten

src/model/d4rt.py:260

↓ 1 callersMethodencode_video

( self, video: torch.Tensor, aspect_ratio: torch.Tensor | None = None, )

src/model/d4rt.py:250

↓ 1 callersFunctionexport_predicted_depth_video

( *, package: dict[str, Any], camera_data: dict[str, np.ndarray], output_dir: Path, grid_r

vis/build_like_demo_for_worldtrack.py:394

↓ 1 callersMethodget_lr

(self)

src/engine/trainer.py:150

↓ 1 callersFunctioniter_tfds_samples

Yield numpy samples from a TFDS split.

src/data_preprocess/prepare_kubric_full_preprocessed.py:294

↓ 1 callersFunctionload_worldtrack_sequence

(npz_path: Path, num_frames: int)

eval_track3d_in_worldtrack.py:57

↓ 1 callersFunctionload_worldtrack_sequence

(npz_path: Path, num_frames: int)

vis/build_like_demo_for_worldtrack.py:120

↓ 1 callersFunctionmain

()

eval_track3d_in_worldtrack.py:434

↓ 1 callersFunctionmain

()

vis/build_like_demo_for_worldtrack.py:762

← previous next →401–500 of 626, ranked by callers