Fix incorrect intensity augmentation defaults in training profiles

gitttt-1234 · claude · gitttt-1234 · commit 2a31bb46d901 · 2026-03-09T10:18:42.000-07:00
The training profile YAML files had legacy/inconsistent values for intensity augmentation parameters. These are now standardized to match sleap-nn PR #485. Updated all 9 training profiles with correct normalized values: | Parameter | Old | New | |-----------|-----|-----| | gaussian_noise_mean | 5.0 | 0.0 | | gaussian_noise_std | 1.0 | 0.02 | | uniform_noise_max | 1.0 | 0.04 | | contrast_min | 0.5 | 0.9 | | contrast_max | 2.0 | 1.1 | | brightness_min | 0.0 | 0.9 | | brightness_max | 2.0 | 1.1 | The old values would produce unusable results if augmentations were enabled: - gaussian_noise_mean=5.0 → 5.0*255=1275 (outside uint8 range) - brightness_min=0.0 → completely black images Fixes #2643 Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
diff --git a/sleap/training_profiles/baseline.centroid.yaml b/sleap/training_profiles/baseline.centroid.yaml
@@ -21,16 +21,16 @@ data_config:
   augmentation_config:
     intensity:
       uniform_noise_min: 0.0
-      uniform_noise_max: 1.0
+      uniform_noise_max: 0.04
       uniform_noise_p: 0.0
-      gaussian_noise_mean: 5.0
-      gaussian_noise_std: 1.0
+      gaussian_noise_mean: 0.0
+      gaussian_noise_std: 0.02
       gaussian_noise_p: 0.0
-      contrast_min: 0.5
-      contrast_max: 2.0
+      contrast_min: 0.9
+      contrast_max: 1.1
       contrast_p: 0.0
-      brightness_min: 0.0
-      brightness_max: 2.0
+      brightness_min: 0.9
+      brightness_max: 1.1
       brightness_p: 0.0
     geometric:
       rotation_min: -180.0
diff --git a/sleap/training_profiles/baseline.multi_class_bottomup.yaml b/sleap/training_profiles/baseline.multi_class_bottomup.yaml
@@ -21,10 +21,10 @@ data_config:
   augmentation_config:
     intensity:
       uniform_noise_min: 0.0
-      uniform_noise_max: 1.0
+      uniform_noise_max: 0.04
       uniform_noise_p: 0.0
-      gaussian_noise_mean: 5.0
-      gaussian_noise_std: 1.0
+      gaussian_noise_mean: 0.0
+      gaussian_noise_std: 0.02
       gaussian_noise_p: 0.0
       contrast_min: 0.9
       contrast_max: 1.1
diff --git a/sleap/training_profiles/baseline.multi_class_topdown.yaml b/sleap/training_profiles/baseline.multi_class_topdown.yaml
@@ -21,10 +21,10 @@ data_config:
   augmentation_config:
     intensity:
       uniform_noise_min: 0.0
-      uniform_noise_max: 1.0
+      uniform_noise_max: 0.04
       uniform_noise_p: 0.0
-      gaussian_noise_mean: 5.0
-      gaussian_noise_std: 1.0
+      gaussian_noise_mean: 0.0
+      gaussian_noise_std: 0.02
       gaussian_noise_p: 0.0
       contrast_min: 0.9
       contrast_max: 1.1
diff --git a/sleap/training_profiles/baseline_large_rf.bottomup.yaml b/sleap/training_profiles/baseline_large_rf.bottomup.yaml
@@ -21,16 +21,16 @@ data_config:
   augmentation_config:
     intensity:
       uniform_noise_min: 0.0
-      uniform_noise_max: 1.0
+      uniform_noise_max: 0.04
       uniform_noise_p: 0.0
-      gaussian_noise_mean: 5.0
-      gaussian_noise_std: 1.0
+      gaussian_noise_mean: 0.0
+      gaussian_noise_std: 0.02
       gaussian_noise_p: 0.0
-      contrast_min: 0.5
-      contrast_max: 2.0
+      contrast_min: 0.9
+      contrast_max: 1.1
       contrast_p: 0.0
-      brightness_min: 0.0
-      brightness_max: 2.0
+      brightness_min: 0.9
+      brightness_max: 1.1
       brightness_p: 0.0
     geometric:
       rotation_min: -180.0
diff --git a/sleap/training_profiles/baseline_large_rf.single.yaml b/sleap/training_profiles/baseline_large_rf.single.yaml
@@ -21,16 +21,16 @@ data_config:
   augmentation_config:
     intensity:
       uniform_noise_min: 0.0
-      uniform_noise_max: 1.0
+      uniform_noise_max: 0.04
       uniform_noise_p: 0.0
-      gaussian_noise_mean: 5.0
-      gaussian_noise_std: 1.0
+      gaussian_noise_mean: 0.0
+      gaussian_noise_std: 0.02
       gaussian_noise_p: 0.0
-      contrast_min: 0.5
-      contrast_max: 2.0
+      contrast_min: 0.9
+      contrast_max: 1.1
       contrast_p: 0.0
-      brightness_min: 0.0
-      brightness_max: 2.0
+      brightness_min: 0.9
+      brightness_max: 1.1
       brightness_p: 0.0
     geometric:
       rotation_min: -180.0
diff --git a/sleap/training_profiles/baseline_large_rf.topdown.yaml b/sleap/training_profiles/baseline_large_rf.topdown.yaml
@@ -21,16 +21,16 @@ data_config:
   augmentation_config:
     intensity:
       uniform_noise_min: 0.0
-      uniform_noise_max: 1.0
+      uniform_noise_max: 0.04
       uniform_noise_p: 0.0
-      gaussian_noise_mean: 5.0
-      gaussian_noise_std: 1.0
+      gaussian_noise_mean: 0.0
+      gaussian_noise_std: 0.02
       gaussian_noise_p: 0.0
-      contrast_min: 0.5
-      contrast_max: 2.0
+      contrast_min: 0.9
+      contrast_max: 1.1
       contrast_p: 0.0
-      brightness_min: 0.0
-      brightness_max: 2.0
+      brightness_min: 0.9
+      brightness_max: 1.1
       brightness_p: 0.0
     geometric:
       rotation_min: -180.0
diff --git a/sleap/training_profiles/baseline_medium_rf.bottomup.yaml b/sleap/training_profiles/baseline_medium_rf.bottomup.yaml
@@ -21,16 +21,16 @@ data_config:
   augmentation_config:
     intensity:
       uniform_noise_min: 0.0
-      uniform_noise_max: 1.0
+      uniform_noise_max: 0.04
       uniform_noise_p: 0.0
-      gaussian_noise_mean: 5.0
-      gaussian_noise_std: 1.0
+      gaussian_noise_mean: 0.0
+      gaussian_noise_std: 0.02
       gaussian_noise_p: 0.0
-      contrast_min: 0.5
-      contrast_max: 2.0
+      contrast_min: 0.9
+      contrast_max: 1.1
       contrast_p: 0.0
-      brightness_min: 0.0
-      brightness_max: 2.0
+      brightness_min: 0.9
+      brightness_max: 1.1
       brightness_p: 0.0
     geometric:
       rotation_min: -180.0
diff --git a/sleap/training_profiles/baseline_medium_rf.single.yaml b/sleap/training_profiles/baseline_medium_rf.single.yaml
@@ -21,16 +21,16 @@ data_config:
   augmentation_config:
     intensity:
       uniform_noise_min: 0.0
-      uniform_noise_max: 1.0
+      uniform_noise_max: 0.04
       uniform_noise_p: 0.0
-      gaussian_noise_mean: 5.0
-      gaussian_noise_std: 1.0
+      gaussian_noise_mean: 0.0
+      gaussian_noise_std: 0.02
       gaussian_noise_p: 0.0
-      contrast_min: 0.5
-      contrast_max: 2.0
+      contrast_min: 0.9
+      contrast_max: 1.1
       contrast_p: 0.0
-      brightness_min: 0.0
-      brightness_max: 2.0
+      brightness_min: 0.9
+      brightness_max: 1.1
       brightness_p: 0.0
     geometric:
       rotation_min: -180.0
diff --git a/sleap/training_profiles/baseline_medium_rf.topdown.yaml b/sleap/training_profiles/baseline_medium_rf.topdown.yaml
@@ -21,16 +21,16 @@ data_config:
   augmentation_config:
     intensity:
       uniform_noise_min: 0.0
-      uniform_noise_max: 1.0
+      uniform_noise_max: 0.04
       uniform_noise_p: 0.0
-      gaussian_noise_mean: 5.0
-      gaussian_noise_std: 1.0
+      gaussian_noise_mean: 0.0
+      gaussian_noise_std: 0.02
       gaussian_noise_p: 0.0
-      contrast_min: 0.5
-      contrast_max: 2.0
+      contrast_min: 0.9
+      contrast_max: 1.1
       contrast_p: 0.0
-      brightness_min: 0.0
-      brightness_max: 2.0
+      brightness_min: 0.9
+      brightness_max: 1.1
       brightness_p: 0.0
     geometric:
       rotation_min: -180.0
