Types & classes — github.com/Lijiaxin0111/Open-d4rt

CodeHub MCPcopy Create free account

hub / github.com/Lijiaxin0111/Open-d4rt / types & classes

Types & classes62 in github.com/Lijiaxin0111/Open-d4rt

⨍ Functions 626◇ Types & classes 62

↓ 53 callersClassRetryableSampleError

Exception type for data-corruption/read errors that should be retried.

src/data/bad_sample_registry.py:17

↓ 21 callersClassRawAugmentConfig

src/data/raw_augment.py:17

↓ 10 callersClassBadSampleRegistry

Process-safe JSON registry for bad samples and bad paths.

src/data/bad_sample_registry.py:56

↓ 4 callersClassConfigNode

Dictionary with attribute-style access.

src/core/config.py:22

↓ 3 callersClassRegistry

Named registry for extensible builders.

src/core/registry.py:8

↓ 3 callersClass_FrameCamera

src/data/dynamic_replica_raw_dataset.py:193

↓ 2 callersClassSourceRecord

src/data_preprocess/prepare_kubric_full_preprocessed.py:79

↓ 2 callersClassStepEvalRecord

scripts/eval_worldtrack/batch_eval_worldtrack_step_ckpts.py:110

↓ 1 callersClassBlendermvsRawConfig

src/data/blendermvs_raw_dataset.py:143

↓ 1 callersClassBlendermvsRawDataset

Loads BlenderMVS scene folders and builds D4RT-compatible supervision.

src/data/blendermvs_raw_dataset.py:192

↓ 1 callersClassCo3dRawConfig

src/data/co3d_raw_dataset.py:125

↓ 1 callersClassCo3dRawDataset

Loads CO3D v2 from raw files and builds D4RT-compatible supervision.

src/data/co3d_raw_dataset.py:169

↓ 1 callersClassCrossAttentionBlock

Decoder block with cross-attention only (no query self-attention).

src/model/decoder.py:11

↓ 1 callersClassD4RTHeads

Predicts query-level outputs for all configured tasks.

src/model/heads.py:9

↓ 1 callersClassD4RTLoss

src/losses/d4rt_loss.py:13

↓ 1 callersClassD4RTModel

Unified model: encoder + independent query decoder + task heads.

src/model/d4rt.py:169

↓ 1 callersClassDynamicReplicaRawConfig

src/data/dynamic_replica_raw_dataset.py:164

↓ 1 callersClassDynamicReplicaRawDataset

Builds D4RT samples from Dynamic Replica with explicit t_cam geometry transforms.

src/data/dynamic_replica_raw_dataset.py:212

↓ 1 callersClassFourierFeatures

Fourier feature mapping for normalized UV coordinates.

src/model/query_embedding.py:10

↓ 1 callersClassIndependentQueryDecoder

Stack of cross-attention blocks.

src/model/decoder.py:43

↓ 1 callersClassKubricFullRobustConfig

src/data/kubric_full_robust_dataset.py:150

↓ 1 callersClassKubricFullRobustDataset

Loads Kubric MOVi-F full annotations from TFDS and builds robust 3D supervision.

src/data/kubric_full_robust_dataset.py:172

↓ 1 callersClassKubricFullRobustPreprocessConfig

src/data/kubric_full_robust_preprocess_dataset.py:37

↓ 1 callersClassKubricFullRobustPreprocessDataset

Loads preprocessed Kubric MOVi-F scenes and reuses robust supervision construction.

src/data/kubric_full_robust_preprocess_dataset.py:59

↓ 1 callersClassMetricLogger

Append metrics as JSON lines.

src/core/logging.py:32

↓ 1 callersClassMixtureDataset

Round-robin weighted mixture over multiple datasets.

src/data/mixture_dataset.py:39

↓ 1 callersClassMixtureDatasetConfig

src/data/mixture_dataset.py:33

↓ 1 callersClassMvsSynthRawConfig

src/data/mvs_synth_raw_dataset.py:112

↓ 1 callersClassMvsSynthRawDataset

Loads MVS-Synth clips from images/depths/poses and emits D4RT-compatible batches.

src/data/mvs_synth_raw_dataset.py:161

↓ 1 callersClassPointOdysseyRawConfig

src/data/pointodyssey_raw_dataset.py:64

↓ 1 callersClassPointOdysseyRawDataset

Loads PointOdyssey raw files and builds D4RT-compatible supervision.

src/data/pointodyssey_raw_dataset.py:109

↓ 1 callersClassQueryEmbedder

Builds per-query token from UV/time and local RGB patch.

src/model/query_embedding.py:42

↓ 1 callersClassQueryPredictionVisualizer

Periodic query prediction visualizer for train/val loop.

src/vis/training_visualizer.py:324

↓ 1 callersClassSavedScene

src/data_preprocess/prepare_kubric_full_preprocessed.py:64

↓ 1 callersClassScannetRawConfig

src/data/scannet_raw_dataset.py:268

↓ 1 callersClassScannetRawDataset

Loads ScanNet scenes with dense iPhone RGBD supervision when available.

src/data/scannet_raw_dataset.py:418

↓ 1 callersClassSelfAttentionBlock

Pre-norm transformer block with self-attention + MLP.

src/model/encoder.py:27

↓ 1 callersClassTartanairRawConfig

src/data/tartanair_raw_dataset.py:118

↓ 1 callersClassTartanairRawDataset

Loads TartanAir V2 RGBD + poses and builds depth-projected supervision.

src/data/tartanair_raw_dataset.py:153

↓ 1 callersClassTrainer

src/engine/trainer.py:188

↓ 1 callersClassVideoPatchTransformerEncoder

Patchify video then encode tokens with interleaved local/global attention.

src/model/encoder.py:57

↓ 1 callersClassVirtualKitti2RawConfig

src/data/virtual_kitti2_raw_dataset.py:400

↓ 1 callersClassVirtualKitti2RawDataset

Loads Virtual KITTI 2 RGBD + calibration and builds depth-projected supervision.

src/data/virtual_kitti2_raw_dataset.py:436

↓ 1 callersClassWarmupCosineScheduler

src/engine/trainer.py:136

↓ 1 callersClass_DSLRFrameObs

src/data/scannet_raw_dataset.py:287

↓ 1 callersClass_DSLRScene

src/data/scannet_raw_dataset.py:296

↓ 1 callersClass_FilteredDepthSampleError

Sample-level quality filter; retry without persisting a bad-sample entry.

src/data/blendermvs_raw_dataset.py:188

↓ 1 callersClass_FilteredDepthSampleError

Sample-level quality filter; retry without persisting a bad-sample entry.

src/data/mvs_synth_raw_dataset.py:157

↓ 1 callersClass_FrameMeta

src/data/blendermvs_raw_dataset.py:170

↓ 1 callersClass_FrameMeta

src/data/mvs_synth_raw_dataset.py:139

↓ 1 callersClass_FrameMeta

src/data/co3d_raw_dataset.py:147

↓ 1 callersClass_IphoneScene

src/data/scannet_raw_dataset.py:307

↓ 1 callersClass_Scene

src/data/dynamic_replica_raw_dataset.py:204

↓ 1 callersClass_Scene

src/data/pointodyssey_raw_dataset.py:85

↓ 1 callersClass_Scene

src/data/blendermvs_raw_dataset.py:180

↓ 1 callersClass_Scene

src/data/virtual_kitti2_raw_dataset.py:421

↓ 1 callersClass_Scene

src/data/mvs_synth_raw_dataset.py:149

↓ 1 callersClass_Scene

src/data/co3d_raw_dataset.py:160

↓ 1 callersClass_SceneCache

src/data/pointodyssey_raw_dataset.py:96

↓ 1 callersClass_Trajectory

src/data/tartanair_raw_dataset.py:142

ClassQuerySampler

src/data/query_sampler.py:11

ClassSeededDatasetMixin

Utility mixin for deterministic, worker-aware per-sample RNG.

src/data/seeding.py:51