Update kelp species model for RGB imagery

HakaiInstitute · Jul 23, 2024 · 0588bab · 0588bab
1 parent b414e94
commit 0588bab
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/kelp_o_matic/models.py b/kelp_o_matic/models.py
@@ -107,7 +107,9 @@ class KelpRGBSpeciesSegmentationModel(_SpeciesSegmentationModel):
     register_depth = 3
     all_black_val = 0
 
-    torchscript_path = "LRASPP_MobileNetV3_kelp_species_rgb_jit_miou=0.9634.pt"
+    torchscript_path = (
+        "UNetPlusPlus_EfficientNetV2_m_kelp_species_rgb_jit_dice=0.9881.pt"
+    )
     presence_model_class = KelpRGBPresenceSegmentationModel
 
     def post_process(self, x: "torch.Tensor") -> "np.ndarray":