Skip to content

Quick Start

Use this page for the shortest path from setup to first train/inference run.

Environment & Dependencies

  • Python: 3.12.3
  • Install runtime dependencies:
/work/envs/depth/bin/python -m pip install -r requirements.txt
  • Optional docs dependencies:
/work/envs/depth/bin/python -m pip install -r docs/requirements.txt

Quick Training

EO-conditioned multiband training:

/work/envs/depth/bin/python train.py \
  --data-config configs/data_config_eo_4band.yaml \
  --train-config configs/training_config_eo_4band.yaml \
  --model-config configs/model_config_eo_4band.yaml

Legacy single-band training:

/work/envs/depth/bin/python train.py \
  --data-config configs/older_configs/data_config.yaml \
  --train-config configs/older_configs/training_config.yaml \
  --model-config configs/older_configs/model_config.yaml

Quick Inference

Set config/checkpoint constants at the top of inference.py, then run:

/work/envs/depth/bin/python inference.py

For EO multiband runs, use: - MODEL_CONFIG_PATH = "configs/model_config_eo_4band.yaml" - DATA_CONFIG_PATH = "configs/data_config_eo_4band.yaml" - TRAIN_CONFIG_PATH = "configs/training_config_eo_4band.yaml"