Skip to content

Commit 34cc3ce

Browse files
gitttt-1234claude
andcommitted
Fix incorrect gaussian_noise default values in training profiles
The training profile YAML files had legacy values (mean=5.0, std=1.0) from old TensorFlow-based SLEAP. The sleap-nn augmentation code expects values on a 0-1 normalized scale and multiplies by 255 internally. The old values would result in completely washed-out images if gaussian noise was enabled. Updated all 9 training profiles to use correct normalized values: - gaussian_noise_mean: 0.0 (was 5.0) - gaussian_noise_std: 0.02 (was 1.0) Fixes #2643 Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
1 parent 83e38a0 commit 34cc3ce

9 files changed

+18
-18
lines changed

sleap/training_profiles/baseline.centroid.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ data_config:
2323
uniform_noise_min: 0.0
2424
uniform_noise_max: 1.0
2525
uniform_noise_p: 0.0
26-
gaussian_noise_mean: 5.0
27-
gaussian_noise_std: 1.0
26+
gaussian_noise_mean: 0.0
27+
gaussian_noise_std: 0.02
2828
gaussian_noise_p: 0.0
2929
contrast_min: 0.5
3030
contrast_max: 2.0

sleap/training_profiles/baseline.multi_class_bottomup.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ data_config:
2323
uniform_noise_min: 0.0
2424
uniform_noise_max: 1.0
2525
uniform_noise_p: 0.0
26-
gaussian_noise_mean: 5.0
27-
gaussian_noise_std: 1.0
26+
gaussian_noise_mean: 0.0
27+
gaussian_noise_std: 0.02
2828
gaussian_noise_p: 0.0
2929
contrast_min: 0.9
3030
contrast_max: 1.1

sleap/training_profiles/baseline.multi_class_topdown.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ data_config:
2323
uniform_noise_min: 0.0
2424
uniform_noise_max: 1.0
2525
uniform_noise_p: 0.0
26-
gaussian_noise_mean: 5.0
27-
gaussian_noise_std: 1.0
26+
gaussian_noise_mean: 0.0
27+
gaussian_noise_std: 0.02
2828
gaussian_noise_p: 0.0
2929
contrast_min: 0.9
3030
contrast_max: 1.1

sleap/training_profiles/baseline_large_rf.bottomup.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ data_config:
2323
uniform_noise_min: 0.0
2424
uniform_noise_max: 1.0
2525
uniform_noise_p: 0.0
26-
gaussian_noise_mean: 5.0
27-
gaussian_noise_std: 1.0
26+
gaussian_noise_mean: 0.0
27+
gaussian_noise_std: 0.02
2828
gaussian_noise_p: 0.0
2929
contrast_min: 0.5
3030
contrast_max: 2.0

sleap/training_profiles/baseline_large_rf.single.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ data_config:
2323
uniform_noise_min: 0.0
2424
uniform_noise_max: 1.0
2525
uniform_noise_p: 0.0
26-
gaussian_noise_mean: 5.0
27-
gaussian_noise_std: 1.0
26+
gaussian_noise_mean: 0.0
27+
gaussian_noise_std: 0.02
2828
gaussian_noise_p: 0.0
2929
contrast_min: 0.5
3030
contrast_max: 2.0

sleap/training_profiles/baseline_large_rf.topdown.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ data_config:
2323
uniform_noise_min: 0.0
2424
uniform_noise_max: 1.0
2525
uniform_noise_p: 0.0
26-
gaussian_noise_mean: 5.0
27-
gaussian_noise_std: 1.0
26+
gaussian_noise_mean: 0.0
27+
gaussian_noise_std: 0.02
2828
gaussian_noise_p: 0.0
2929
contrast_min: 0.5
3030
contrast_max: 2.0

sleap/training_profiles/baseline_medium_rf.bottomup.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ data_config:
2323
uniform_noise_min: 0.0
2424
uniform_noise_max: 1.0
2525
uniform_noise_p: 0.0
26-
gaussian_noise_mean: 5.0
27-
gaussian_noise_std: 1.0
26+
gaussian_noise_mean: 0.0
27+
gaussian_noise_std: 0.02
2828
gaussian_noise_p: 0.0
2929
contrast_min: 0.5
3030
contrast_max: 2.0

sleap/training_profiles/baseline_medium_rf.single.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ data_config:
2323
uniform_noise_min: 0.0
2424
uniform_noise_max: 1.0
2525
uniform_noise_p: 0.0
26-
gaussian_noise_mean: 5.0
27-
gaussian_noise_std: 1.0
26+
gaussian_noise_mean: 0.0
27+
gaussian_noise_std: 0.02
2828
gaussian_noise_p: 0.0
2929
contrast_min: 0.5
3030
contrast_max: 2.0

sleap/training_profiles/baseline_medium_rf.topdown.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ data_config:
2323
uniform_noise_min: 0.0
2424
uniform_noise_max: 1.0
2525
uniform_noise_p: 0.0
26-
gaussian_noise_mean: 5.0
27-
gaussian_noise_std: 1.0
26+
gaussian_noise_mean: 0.0
27+
gaussian_noise_std: 0.02
2828
gaussian_noise_p: 0.0
2929
contrast_min: 0.5
3030
contrast_max: 2.0

0 commit comments

Comments
 (0)