test: update kmeans_segmentation call and adjust expected ice mask sum

WilhelmusLab · Nov 20, 2024 · 55a752a · 55a752a
1 parent a0c7081
commit 55a752a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test-get-ice-masks.jl b/test/test-get-ice-masks.jl
@@ -74,11 +74,11 @@ ref_image_landmasked = apply_landmask(ref_image, .!landmask)
     end
 
     begin
-        morph_residue_seglabels = kmeans_segmentation(Gray.(morph_residue[tile...] / 255))
+        morph_residue_seglabels = kmeans_segmentation(Gray.(morph_residue[tile...] / 255), k=3)
         @test get_nlabel(ref_image_landmasked[tile...], morph_residue_seglabels, 255) == 1
     end
 
     ice_mask, binarized_tiling = get_ice_masks(ref_image, morph_residue, landmask, tiles)
-    @test sum(ice_mask) == 1323384
+    @test sum(ice_mask) == 2669451
     @test sum(binarized_tiling) == 2873080
 end