From 931225a85cbf51706af43702f564992806d8ec95 Mon Sep 17 00:00:00 2001 From: Taylor Denouden Date: Fri, 3 May 2024 15:06:57 -0700 Subject: [PATCH] Mussel bg areas -> 0 --- kelp_o_matic/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kelp_o_matic/models.py b/kelp_o_matic/models.py index f8d99c7..1ccaa8e 100644 --- a/kelp_o_matic/models.py +++ b/kelp_o_matic/models.py @@ -99,7 +99,7 @@ class KelpRGBSpeciesSegmentationModel(_SpeciesSegmentationModel): class MusselRGBPresenceSegmentationModel(_Model): register_depth = 1 - all_black_val = -1 + all_black_val = 0 torchscript_path = ( "UNetPlusPlus_EfficientNetB4_mussel_presence_rgb_jit_dice=0.9269.pt"