Fix dark water threshold: std 0.08 → 0.18
GTA-UAV dark ocean images have std ~0.10-0.15 (blue gradient), not <0.08 as assumed. Raised dark_water_std_thr to 0.18 to cover all dark water images while staying safe for non-water backgrounds. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -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,
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user