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

↓ 2 callersMethodtrain

(self)

src/engine/trainer.py:899

↓ 1 callersMethod__init__

(self, hidden_dim: int, num_heads: int, mlp_ratio: float, dropout: float = 0.1)

src/model/decoder.py:14

↓ 1 callersMethod__init__

( self, hidden_dim: int, clip_frames: int, local_patch_enabled: bool,

src/model/query_embedding.py:45

↓ 1 callersMethod__init__

(self, hidden_dim: int, num_heads: int, mlp_ratio: float, dropout: float = 0.1)

src/model/encoder.py:30

↓ 1 callersMethod__init__

(self, path: str | Path, refresh_seconds: float = 1.0)

src/data/bad_sample_registry.py:59

↓ 1 callersFunction_aggregate_results

(results: list[dict[str, Any]])

eval_track3d_in_worldtrack.py:379

↓ 1 callersFunction_align_model_input_with_train_data

(model_cfg: Any, train_cfg: Any)

train.py:248

↓ 1 callersFunction_align_pred_xyz_sequence

( pred_xyz: np.ndarray, gt_xyz: np.ndarray, mask_xyz: np.ndarray, mode: str, min_points: i

src/vis/training_visualizer.py:123

↓ 1 callersFunction_annotation_filename

(split: str)

src/data/dynamic_replica_raw_dataset.py:102

↓ 1 callersFunction_apply_default_per_dataset_validation

(train_cfg: Any)

train.py:285

↓ 1 callersFunction_apply_sim3_to_pose_seq

(t_ref_cam: np.ndarray, scale: float, rot: np.ndarray, trans: np.ndarray)

vis/build_like_demo.py:816

↓ 1 callersMethod_batch_meta_summary

(self, batch: Any)

src/engine/trainer.py:335

↓ 1 callersFunction_batch_size

(value: Any)

src/engine/trainer.py:66

↓ 1 callersFunction_best_step_suffix

(step: int)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:298

↓ 1 callersMethod_bind_sample_rng

(self, index: int, attempt: int = 0, stream: int = 0)

src/data/seeding.py:87

↓ 1 callersFunction_blendermvs_roots

(cfg: Any)

src/data/builder.py:196

↓ 1 callersFunction_bool_cfg

(value: Any, default: bool = False)

src/engine/trainer.py:123

↓ 1 callersMethod_build_benchmark_tracking

( self, *, scene: _Scene, traj_2d: np.ndarray, traj_3d_world: np.ndarr

src/data/dynamic_replica_raw_dataset.py:270

↓ 1 callersMethod_build_benchmark_tracking

( self, *, depth_z: np.ndarray, depth_range_m: np.ndarray, depth_valid

src/data/kubric_full_robust_dataset.py:675

↓ 1 callersMethod_build_block_modes

(self, num_layers: int)

src/model/encoder.py:96

↓ 1 callersFunction_build_depth_chunk_index

(path: Path)

src/data/scannet_raw_dataset.py:208

↓ 1 callersFunction_build_eval_command

(record: StepEvalRecord, args: argparse.Namespace)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:508

↓ 1 callersFunction_build_meta

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

src/data_preprocess/prepare_kubric_full_preprocessed.py:180

↓ 1 callersFunction_build_object_local_to_world

(bboxes_3d_ot83: np.ndarray)

vis/build_like_demo.py:438

↓ 1 callersFunction_build_per_dataset_val_loaders

(train_cfg: Any, logger: Any, rank: int, world_size: int)

train.py:336

↓ 1 callersMethod_build_queries

( self, video_t_chw: np.ndarray, depth_z: np.ndarray, depth_range_m: np.ndarra

src/data/kubric_full_robust_dataset.py:852

↓ 1 callersFunction_build_query_tensors

( *, query_uv_norm: np.ndarray, t_src: np.ndarray, t_tgt: np.ndarray, t_cam: np.ndarray,

vis/build_like_demo.py:755

↓ 1 callersFunction_build_reference_val_loader

( train_cfg: Any, logger: Any, rank: int, world_size: int, manifest_arg: str | None, )

train.py:311

↓ 1 callersMethod_build_sample

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

src/data/dynamic_replica_raw_dataset.py:597

↓ 1 callersMethod_build_sample

(self, sample: dict[str, Any], idxs: list[int], clip_start: int)

src/data/kubric_full_robust_dataset.py:1039

↓ 1 callersMethod_build_sample

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

src/data/pointodyssey_raw_dataset.py:300

↓ 1 callersMethod_build_sample

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

src/data/blendermvs_raw_dataset.py:383

↓ 1 callersMethod_build_sample

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

src/data/virtual_kitti2_raw_dataset.py:592

↓ 1 callersMethod_build_sample

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

src/data/tartanair_raw_dataset.py:315

↓ 1 callersMethod_build_sample

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

src/data/mvs_synth_raw_dataset.py:313

↓ 1 callersFunction_build_uv_grid

(width: int, height: int, cols: int, rows: int, max_points: int)

vis/build_like_demo.py:104

↓ 1 callersFunction_build_vkitti2_queries_from_trajectories

VKITTI2-specific trajectory builder with pid-aware target-time sampling. The shared builder samples ``t_tgt`` uniformly before choosing a source

src/data/virtual_kitti2_raw_dataset.py:71

↓ 1 callersMethod_camera_mats_and_depth

( self, sample: dict[str, Any], idxs: list[int], )

src/data/kubric_full_robust_dataset.py:469

↓ 1 callersFunction_candidate_pretrained_keys

(target_key: str)

src/model/d4rt.py:52

↓ 1 callersMethod_categories

(self)

src/data/co3d_raw_dataset.py:213

↓ 1 callersMethod_collect_batch_supervision_stats

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

src/engine/trainer.py:585

↓ 1 callersFunction_collect_repro_file_entries

(args: argparse.Namespace, train_cfg: Any)

train.py:121

↓ 1 callersFunction_colorize_depth_map

(depth_hw: np.ndarray, vmin: float, vmax: float)

vis/build_like_demo_for_worldtrack.py:382

↓ 1 callersMethod_compute_disp_loss

( self, outputs: dict[str, torch.Tensor], target: dict[str, torch.Tensor], mas

src/losses/d4rt_loss.py:230

↓ 1 callersFunction_compute_non_boundary_candidate_mask

( *, query_uv_px: np.ndarray, xyz_ref0_frame0: np.ndarray, visibility_frame0: np.ndarray,

vis/build_like_demo.py:134

↓ 1 callersMethod_compute_normal_loss

( self, outputs: dict[str, torch.Tensor], target: dict[str, torch.Tensor], mas

src/losses/d4rt_loss.py:245

↓ 1 callersMethod_compute_reprojection_uv_from_xyz_loss

( self, outputs: dict[str, torch.Tensor], batch: dict[str, Any], metrics: dict

src/losses/d4rt_loss.py:161

↓ 1 callersFunction_compute_scale_factor_global

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

eval_track3d_in_worldtrack.py:131

↓ 1 callersFunction_compute_scale_factor_global

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

vis/build_like_demo_for_worldtrack.py:174

↓ 1 callersMethod_compute_uv_loss

( self, outputs: dict[str, torch.Tensor], target: dict[str, torch.Tensor], mas

src/losses/d4rt_loss.py:91

↓ 1 callersMethod_compute_vis_loss

( self, outputs: dict[str, torch.Tensor], target: dict[str, torch.Tensor], mas

src/losses/d4rt_loss.py:215

↓ 1 callersMethod_compute_xyz_loss

( self, outputs: dict[str, torch.Tensor], target: dict[str, torch.Tensor], mas

src/losses/d4rt_loss.py:28

↓ 1 callersMethod_configured_roots

(self)

src/data/blendermvs_raw_dataset.py:228

↓ 1 callersFunction_count_eval_cases

(data_root: Path, subsets_csv: str, limit_seqs: int)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:81

↓ 1 callersFunction_count_split_examples

(input_root: Path, split: str)

src/data_preprocess/prepare_kubric_full_preprocessed.py:310

↓ 1 callersFunction_debug_scalar

(value: Any)

src/engine/trainer.py:51

↓ 1 callersMethod_decode_depth

(self, depth_u16: np.ndarray)

src/data/dynamic_replica_raw_dataset.py:385

↓ 1 callersFunction_decode_depth_chunk

Decode one ScanNet++ iPhone depth chunk to depth in meters. In practice, scenes can mix at least two encodings: - lz4.block compressed uint16

src/data/scannet_raw_dataset.py:226

↓ 1 callersFunction_decode_depth_u16_to_metric

(depth_u16: np.ndarray, depth_range: np.ndarray)

vis/build_like_demo.py:427

↓ 1 callersFunction_decode_depth_u16_to_metric

(depth_u16: np.ndarray, depth_range: np.ndarray)

src/data/kubric_full_robust_dataset.py:94

↓ 1 callersFunction_decode_jpeg_rgb

(frame_bytes: bytes)

eval_track3d_in_worldtrack.py:38

↓ 1 callersFunction_decode_jpeg_rgb

(frame_bytes: bytes)

vis/build_like_demo_for_worldtrack.py:87

↓ 1 callersFunction_decode_object_coordinates_u16

(obj_u16: np.ndarray)

vis/build_like_demo.py:434

↓ 1 callersFunction_decode_object_coordinates_u16

(obj_u16: np.ndarray)

src/data/kubric_full_robust_dataset.py:101

↓ 1 callersFunction_decode_text

(value: Any)

src/data_preprocess/prepare_kubric_full_preprocessed.py:94

↓ 1 callersFunction_decode_world_normals_u16

(normal_u16: np.ndarray)

src/data/kubric_full_robust_dataset.py:105

↓ 1 callersFunction_depth_seg_occlusion

( *, depth_z_hw: np.ndarray, depth_valid_hw: np.ndarray, seg_hw: np.ndarray, u: float,

vis/build_like_demo.py:480

↓ 1 callersMethod_detect_non_finite

(self, loss: torch.Tensor, metrics: dict[str, Any])

src/engine/trainer.py:634

↓ 1 callersFunction_disable_encoder_pretrain

(cfg: Any)

vis/build_like_demo_for_worldtrack.py:263

↓ 1 callersMethod_discover_scene_dirs

(self)

src/data/kubric_full_robust_preprocess_dataset.py:93

↓ 1 callersFunction_discover_step_ckpts

(args: argparse.Namespace)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:217

↓ 1 callersFunction_discover_top_summary_ckpts

(args: argparse.Namespace)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:356

↓ 1 callersMethod_display_worldtrack_eval_command

(self, env: dict[str, str], script_arg: str)

src/engine/trainer.py:425

↓ 1 callersMethod_distributed_mean_scalar_dict

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

src/engine/trainer.py:673

↓ 1 callersFunction_dynamic_replica_split

(split: str, cfg: Any)

src/data/builder.py:406

↓ 1 callersMethod_effective_split

(self)

src/data/tartanair_raw_dataset.py:274

↓ 1 callersMethod_empty_payload

(self)

src/data/bad_sample_registry.py:77

↓ 1 callersFunction_encoder_variant_defaults

(variant: str)

src/model/d4rt.py:41

↓ 1 callersMethod_enumerate_trajectories

(self)

src/data/tartanair_raw_dataset.py:186

↓ 1 callersFunction_estimate_ref0_intrinsics

( *, xyz_ref0_frame0: np.ndarray, uv_px_frame0: np.ndarray, visibility_frame0: np.ndarray,

vis/build_like_demo.py:289

↓ 1 callersFunction_estimate_sim3_ransac

( src: np.ndarray, dst: np.ndarray, iterations: int = 1000, inlier_threshold: float = 0.05, )

eval_track3d_in_worldtrack.py:186

↓ 1 callersMethod_eval_clip_slot

(self, index: int)

src/data/pointodyssey_raw_dataset.py:277

↓ 1 callersMethod_eval_sample_by_index

(self, index: int)

src/data/kubric_full_robust_dataset.py:429

↓ 1 callersFunction_existing_64clip_summary

( exp_dir: Path, step: int, args: argparse.Namespace, )

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:310

↓ 1 callersFunction_export_demo_data

( *, model: torch.nn.Module, video_rgb: np.ndarray, video_model_rgb: np.ndarray, point_que

vis/build_like_demo.py:1272

↓ 1 callersFunction_extract_failed_paths_from_error_text

(message: str)

src/data/kubric_full_robust_dataset.py:137

↓ 1 callersMethod_extract_local_patches

( self, video: torch.Tensor, u: torch.Tensor, v: torch.Tensor, t_src:

src/model/query_embedding.py:80

↓ 1 callersFunction_extract_tensorboard_scalars

(summary_json: Path)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:565

↓ 1 callersFunction_figure_to_rgb

(fig: plt.Figure)

vis/build_like_demo_for_worldtrack.py:256

↓ 1 callersFunction_figure_to_rgb

(fig: Any)

src/vis/training_visualizer.py:62

↓ 1 callersMethod_filter_sequence_names_by_quality

(self, category_dir: Path, sequence_names: list[str])

src/data/co3d_raw_dataset.py:262

↓ 1 callersMethod_flatten_per_dataset_metrics

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

src/engine/trainer.py:877

↓ 1 callersFunction_format_stage_times

(stage_times: dict[str, float])

src/engine/trainer.py:98

↓ 1 callersMethod_frame_ids

(self, scene: _Scene, index: int)

src/data/dynamic_replica_raw_dataset.py:551

↓ 1 callersMethod_frame_indices

(self, scene_len: int, index: int)

src/data/blendermvs_raw_dataset.py:364

↓ 1 callersMethod_frame_indices

(self, scene_len: int, index: int)

src/data/virtual_kitti2_raw_dataset.py:572

↓ 1 callersMethod_frame_indices

(self, traj_len: int, index: int)

src/data/tartanair_raw_dataset.py:294

↓ 1 callersMethod_frame_indices

(self, scene_len: int, index: int)

src/data/mvs_synth_raw_dataset.py:294

↓ 1 callersMethod_frame_indices

(self, scene: _Scene, index: int)

src/data/co3d_raw_dataset.py:434

↓ 1 callersFunction_full_eval_best_ckpt_still_matches

(row: dict[str, Any], best_ckpt: Path)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:326

← previous next →201–300 of 626, ranked by callers