diff --git a/src/augmentor/inference.py b/src/augmentor/inference.py index 4869ef1..6f07d0c 100644 --- a/src/augmentor/inference.py +++ b/src/augmentor/inference.py @@ -311,7 +311,7 @@ def postprocess_segmentation( images_raw: torch.Tensor, water_class: int = 4, dark_water_mean_thr: float = 0.24, - dark_water_std_thr: float = 0.08, + dark_water_std_thr: float = 0.18, reclassify_wetland: bool = False, wetland_class: int = 14, vegetation_class: int = 3, diff --git a/src/tests/test_inference.py b/src/tests/test_inference.py index 757a493..d326416 100644 --- a/src/tests/test_inference.py +++ b/src/tests/test_inference.py @@ -230,9 +230,9 @@ class TestInferSegmentationBatch: class TestPostprocessSegmentation: def test_dark_water_reclassified(self) -> None: """Dark uniform background → water.""" - # Dark image (mean ~0.15, std ~0.02) + # Dark image (mean ~0.15, std ~0.10 — typical dark ocean) rgb = torch.full((1, 3, 32, 32), 0.15) - rgb += torch.randn_like(rgb) * 0.02 + rgb += torch.randn_like(rgb) * 0.05 rgb.clamp_(0, 1) seg = torch.zeros(1, 1, 32, 32, dtype=torch.uint8) # all background result = postprocess_segmentation(seg, rgb)