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

↓ 3 callersFunction_step

(delta: int)

vis/serve_demo_viser.py:469

↓ 3 callersMethod_tb_add_scalar

(self, tag: str, value: Any, step: int)

src/engine/trainer.py:682

↓ 3 callersMethod_warn_skip

(self, sample_key: str, reason: str, failed_paths: list[str])

src/data/kubric_full_robust_dataset.py:321

↓ 3 callersFunction_wrap

(value: Any)

src/core/config.py:13

↓ 3 callersFunction_write_tensorboard_for_complete_records

(records: list[StepEvalRecord], *, force: bool = False)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:649

↓ 3 callersFunctionbuild_logger

(name: str, log_dir: str | Path)

src/core/logging.py:11

↓ 3 callersFunctionbuild_model

(model_section_cfg: Any)

src/model/builder.py:19

↓ 3 callersFunctionsave_checkpoint

(path: str | Path, payload: dict[str, Any])

src/core/checkpoint.py:11

↓ 3 callersFunctionseed_everything

(seed: int, deterministic: bool = False)

src/core/seed.py:11

↓ 3 callersFunctionstable_split_bucket

Stable hash bucket for dataset splits across processes and runs.

src/data/seeding.py:34

↓ 2 callersFunction_apply_sim3_to_xyz

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

infer_track_3d.py:158

↓ 2 callersFunction_best_eval_dir

(exp_dir: Path, step: int, eval_dir_suffix: str)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:306

↓ 2 callersFunction_best_mode_for_metric

(metric_name: str, raw_mode: str)

src/engine/trainer.py:110

↓ 2 callersMethod_build_object_local_to_world

( self, bboxes_3d_ot83: np.ndarray, )

src/data/kubric_full_robust_dataset.py:634

↓ 2 callersMethod_build_sample

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

src/data/co3d_raw_dataset.py:454

↓ 2 callersFunction_camera_pose_from_meta

( meta: dict[str, Any], frame_idx: int, camera_source: str, )

vis/serve_demo_viser.py:197

↓ 2 callersFunction_count_completed_cases_from_log

(log_path: Path)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:94

↓ 2 callersFunction_decode_depth_float16_bitcast

(depth_u16: np.ndarray)

src/data/dynamic_replica_raw_dataset.py:53

↓ 2 callersFunction_decode_depth_uint16_divisor

(depth_u16: np.ndarray, depth_divisor: float)

src/data/dynamic_replica_raw_dataset.py:59

↓ 2 callersFunction_depth_decode_plausibility

(depth_m: np.ndarray, eps: float = 1e-6)

src/data/dynamic_replica_raw_dataset.py:67

↓ 2 callersMethod_depth_seg_occlusion

( depth_z_hw: np.ndarray, depth_valid_hw: np.ndarray, seg_hw: np.ndarray, u: f

src/data/kubric_full_robust_dataset.py:594

↓ 2 callersFunction_effective_crop_aspect_ratio

( native_aspect_ratio: np.ndarray, *, crop_h: int, crop_w: int, out_h: int, out_w: int

src/data/raw_augment.py:599

↓ 2 callersFunction_enforce_shared_focal_intrinsics

(intrinsics_params: np.ndarray)

vis/build_like_demo.py:328

↓ 2 callersFunction_estimate_intrinsics_params_from_predictions

( pred_tracks: np.ndarray, pred_uv_norm: np.ndarray, image_hw: tuple[int, int], )

src/eval/tasks.py:72

↓ 2 callersFunction_estimate_overlap_sim3

( *, prev_xyz_qt3: np.ndarray, curr_xyz_qt3: np.ndarray, prev_vis_qt: np.ndarray, curr_vis

infer_track_3d.py:176

↓ 2 callersFunction_estimate_relative_pose_from_queries

( *, model: torch.nn.Module, video_clip: torch.Tensor, aspect_ratio: torch.Tensor | None,

vis/build_like_demo.py:994

↓ 2 callersMethod_event_dir

(self, split: str, step: int)

src/vis/training_visualizer.py:392

↓ 2 callersMethod_finalize_validation_metrics

( self, val_loss_sum: float, val_count: int, val_metric_sums: dict[str, float]

src/engine/trainer.py:823

↓ 2 callersFunction_finite_correspondence_pairs

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

eval_track3d_in_worldtrack.py:221

↓ 2 callersFunction_format_subset_summary

(subset: str, summary: dict[str, Any])

eval_track3d_in_worldtrack.py:423

↓ 2 callersFunction_fov_from_k

(k: np.ndarray, image_h: int)

vis/serve_demo_viser.py:167

↓ 2 callersMethod_frame_camera

(self, scene: _Scene, frame_idx: int)

src/data/dynamic_replica_raw_dataset.py:564

↓ 2 callersFunction_frame_id_from_filename

(path: Path)

src/data/virtual_kitti2_raw_dataset.py:345

↓ 2 callersMethod_frame_indices

(self, scene_len: int, index: int)

src/data/pointodyssey_raw_dataset.py:283

↓ 2 callersMethod_global_batches_for_sample_limit

(self, loader: Any | None, max_samples_global: int)

src/engine/trainer.py:787

↓ 2 callersFunction_infer_tracks

( *, model: torch.nn.Module, video_model_rgb: np.ndarray, query_uv_norm: np.ndarray, query

infer_track_3d.py:372

↓ 2 callersFunction_jump_client_view_to_camera_pose

( client: Any, t_wc: np.ndarray, *, fov: float | None = None, fallback_look_distance: floa

vis/serve_demo_viser.py:74

↓ 2 callersFunction_line_ids

(num_pairs: int, max_lines: int)

src/vis/training_visualizer.py:30

↓ 2 callersFunction_load_json

(path: Path)

src/data/kubric_full_robust_preprocess_dataset.py:42

↓ 2 callersFunction_load_json_gz

(path: Path)

src/data/co3d_raw_dataset.py:33

↓ 2 callersMethod_load_scene_cached

(self, sid: str)

src/data/scannet_raw_dataset.py:452

↓ 2 callersFunction_load_tfds_modules

()

src/data_preprocess/prepare_kubric_full_preprocessed.py:278

↓ 2 callersFunction_load_traj

(path: Path)

src/data/dynamic_replica_raw_dataset.py:90

↓ 2 callersFunction_loader_int

(split: str, cfg: Any, key: str, default: int | None)

src/data/builder.py:705

↓ 2 callersMethod_log_stage_times

( self, step: int, stage_times: dict[str, float], *, batch: Any | None

src/engine/trainer.py:352

↓ 2 callersMethod_lr_at_step

(self, step_id: int)

src/engine/trainer.py:153

↓ 2 callersFunction_make_sliding_window_clip_ranges

( num_frames: int, clip_frames: int, overlap_frames: int | None = None, )

infer_track_3d.py:139

↓ 2 callersFunction_mark

(diff: np.ndarray, z_ref: np.ndarray, vmask: np.ndarray, sl_a: tuple[slice, slice], sl_b: tuple[slice, slice])

vis/build_like_demo.py:158

↓ 2 callersFunction_maybe_accept

(crop_h: int, crop_w: int)

src/data/raw_augment.py:520

↓ 2 callersFunction_metric_token

(value: str)

src/engine/trainer.py:105

↓ 2 callersFunction_min_max_pair

(value: Any, default: tuple[float, float], *, floor: float)

src/data/raw_augment.py:138

↓ 2 callersFunction_normalize_mixture_name

(name: str)

src/data/builder.py:559

↓ 2 callersMethod_now_iso

(self)

src/data/bad_sample_registry.py:68

↓ 2 callersFunction_parse_row

(row: str, expected: int)

src/data/blendermvs_raw_dataset.py:112

↓ 2 callersFunction_plot_overlay_2d

( image: np.ndarray, gt_xy: np.ndarray, pred_xy: np.ndarray, errors_px: np.ndarray, hard_f

src/vis/training_visualizer.py:175

↓ 2 callersFunction_probability_if_enabled

(cfg: Any, enabled_path: str, probability_path: str)

src/data/raw_augment.py:122

↓ 2 callersMethod_project_point

(k: np.ndarray, t_cw: np.ndarray, p_world: np.ndarray)

src/data/kubric_full_robust_dataset.py:580

↓ 2 callersFunction_project_world_tracks_to_uv

(points_world_tq3: np.ndarray, extrinsics_w2c: np.ndarray, intrinsics: np.ndarray)

vis/build_like_demo_for_worldtrack.py:239

↓ 2 callersFunction_read_json_dict

(path: Path)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:288

↓ 2 callersMethod_refresh_if_needed

(self)

src/data/bad_sample_registry.py:125

↓ 2 callersMethod_reload

(self, force: bool = False)

src/data/bad_sample_registry.py:84

↓ 2 callersFunction_render_track_layer

( *, name_prefix: str, xyz: np.ndarray, vis: np.ndarray, colors: np.nd

vis/serve_demo_viser.py:323

↓ 2 callersMethod_reset_eval_iter

(self)

src/data/kubric_full_robust_dataset.py:282

↓ 2 callersFunction_resize_video

(video_rgb: np.ndarray, image_hw: tuple[int, int])

infer_track_3d.py:37

↓ 2 callersFunction_resolve_device

(raw: str)

infer_track_3d.py:15

↓ 2 callersFunction_resolve_float

(value: Any, default: float)

src/model/d4rt.py:23

↓ 2 callersFunction_rotmat_to_wxyz

(rot: np.ndarray)

vis/serve_demo_viser.py:42

↓ 2 callersFunction_safe_float

(value: Any)

src/engine/trainer.py:41

↓ 2 callersFunction_safe_name

(value: str)

src/vis/training_visualizer.py:20

↓ 2 callersFunction_sample_crop_window

( h: int, w: int, rng: np.random.Generator, cfg: RawAugmentConfig, native_aspect_ratio: fl

src/data/raw_augment.py:486

↓ 2 callersFunction_sample_log_uniform

(lo: float, hi: float)

src/data/raw_augment.py:517

↓ 2 callersMethod_sample_rng

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

src/data/seeding.py:84

↓ 2 callersMethod_scalarize_metric_dict

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

src/engine/trainer.py:664

↓ 2 callersMethod_select_clip

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

src/data/kubric_full_robust_dataset.py:458

↓ 2 callersFunction_select_motion_tracks

( *, query_uv_px: np.ndarray, xyz_ref0: np.ndarray, visibility: np.ndarray, confidence: np

vis/build_like_demo.py:227

↓ 2 callersFunction_split_csv_paths

(raw: str | None)

train.py:94

↓ 2 callersFunction_step_eval_dir

(exp_dir: Path, step: int, eval_dir_suffix: str)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:302

↓ 2 callersFunction_summary_inputs_match

(path: Path, args: argparse.Namespace)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:199

↓ 2 callersFunction_to_device

(value: Any, device: torch.device)

src/engine/trainer.py:25

↓ 2 callersFunction_to_float

(value: Any)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:535

↓ 2 callersFunction_track_colors

(n: int)

vis/serve_demo_viser.py:138

↓ 2 callersFunction_transform_uv

( uv_norm: np.ndarray, x0: int, y0: int, crop_h: int, crop_w: int, h: int, w: int,

src/data/raw_augment.py:578

↓ 2 callersMethod_unproject_background_world

( self, u: int, v: int, depth_range_value: float, k: np.ndarray,

src/data/kubric_full_robust_dataset.py:829

↓ 2 callersFunction_unwrap_state_dict

(payload: Any)

infer_track_3d.py:26

↓ 2 callersMethod_validate_runtime_sample

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

src/data/kubric_full_robust_dataset.py:350

↓ 2 callersMethod_validate_scene_dir

(self, scene_dir: Path)

src/data/kubric_full_robust_preprocess_dataset.py:114

↓ 2 callersMethod_validate_schema

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

src/data/kubric_full_robust_dataset.py:332

↓ 2 callersFunction_worldtrack_scale_from_meta

(meta: dict[str, Any])

vis/serve_demo_viser.py:175

↓ 2 callersFunction_write_tensorboard

(record: StepEvalRecord)

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:624

↓ 2 callersFunction_write_yaml

(path: Path, payload: dict[str, object])

train.py:151

↓ 2 callersMethod_xyz_preprocess

(self, xyz: torch.Tensor, mask: torch.Tensor, normalize_depth: bool, transform_log: bool)

src/losses/d4rt_loss.py:18

↓ 2 callersFunctionapply_overrides

(config: ConfigNode, overrides: list[str] | None)

src/core/config.py:86

↓ 2 callersFunctionbuild_queries_from_trajectories

Build D4RT query/target/mask/query_stats from GT 3D trajectories. Unlike ``build_queries_from_depth`` which assumes static-scene geometry, th

src/data/depth_query_builder.py:277

↓ 2 callersFunctioncount_valid_depth_frames

(depth_valid_t_hw: np.ndarray, *, min_valid_ratio: float)

src/data/raw_augment.py:74

↓ 2 callersFunctionget_worker_info

()

src/data/seeding.py:25

↓ 2 callersMethodlog_batch

(self, batch: dict[str, Any], outputs: dict[str, Any], split: str, step: int)

src/vis/training_visualizer.py:643

↓ 2 callersFunctionrender

()

vis/serve_demo_viser.py:380

↓ 2 callersFunctionsanitize_depth_map

Return a filtered depth map and its validity mask. Invalid values are represented by the returned validity mask; the depth array is set to Na

src/data/raw_augment.py:40

↓ 2 callersMethodsave_last_checkpoint

(self)

src/engine/trainer.py:397

↓ 2 callersMethodstep

(self)

src/engine/trainer.py:161

← previous next →101–200 of 626, ranked by callers