DualEarth · evan-powers · Jun 8, 2026
diff --git a/README.md b/README.md
@@ -11,6 +11,7 @@ The Deep Terrain Routing Lab is aimed at developing ML tools for studying and pr
 This component focuses on creating synthetic datasets for training and validating our terrain routing models:
 
 - **Synthetic Digital Elevation Models (DEM)**: Generate DEMs that represent realistic topographical features.
+     - **Catchment Delineation**: Calculates DEM catchment boundaries (can be toggled on/off in config).
 - **Synthetic Precipitation Layers**: Simulate precipitation events driving surface water response.
 
 ### 2. Terrain Routing Simulation
@@ -20,3 +21,4 @@ Simulate water flow across the synthetic DEMs to represent hydrologic response u
 ### 3. Deep Learning Model for Terrain Routing
 
 Develop a neural network model to predict the distribution of water across the terrain rainfall.
+
diff --git a/config/config.yaml b/config/config.yaml
@@ -48,11 +48,12 @@ routing:
 # ----------------------------
 dataset:
   png_dir: "./dataset/png"
-  num_samples: 5      # number of synthetic scenarios
+  num_samples: 2      # number of synthetic scenarios
   rain_snapshots: 5      # number of rainfall snapshots for ML input
   snapshot_stride: 3    # time gap (timesteps) between each snapshot
   use_momentum_routing: false  # whether to use the ShallowWaterRouter
   apply_cloud_mask: false
+  run_catchment_delineation: true  # if true, run catchment.py after data generation
 
 # ----------------------------
 # Specifications for ML Model

diff --git a/environment/environment_cpu.yml b/environment/environment_cpu.yml
@@ -21,5 +21,6 @@ dependencies:
   - pyyaml
   - pip
   - pip:
+      - pysheds
       - tensorboard
       - torchviz
diff --git a/environment/environment_cuda.yml b/environment/environment_cuda.yml
@@ -21,5 +21,6 @@ dependencies:
   - pyyaml
   - pip
   - pip:
+      - pysheds
       - tensorboard
       - torchviz
diff --git a/environment/environment_mps.yml b/environment/environment_mps.yml
@@ -19,5 +19,6 @@ dependencies:
   - pyyaml
   - pip
   - pip:
+      - pysheds
       - tensorboard
       - torchviz
diff --git a/environment/requirements.txt b/environment/requirements.txt
@@ -1,2 +1,5 @@
 numpy
 scipy
+matplotlib
+plotly
+pysheds
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,7 +12,13 @@ authors = [
     {name = "Jackson Laney", email = "jrlaney1@crimson.ua.edu"}
     ]
 requires-python = ">=3.8"
-dependencies = []
+dependencies = [
+    "matplotlib",
+    "numpy",
+    "plotly",
+    "pysheds",
+    "scipy",
+]
 
 [tool.setuptools.packages.find]
 where = ["src"]
-Original file line number
+Diff line change
@@ Expand Up / @@ -21,5 +21,6 @@ dependencies: @@
       - pyyaml
       - pip
       - pip:
+          - pysheds
           - tensorboard
           - torchviz