From 0588bab6c7ab9078ae3778e8237356b75bd25288 Mon Sep 17 00:00:00 2001 From: Taylor Denouden Date: Tue, 23 Jul 2024 12:00:23 -0700 Subject: [PATCH] Update kelp species model for RGB imagery --- kelp_o_matic/models.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/kelp_o_matic/models.py b/kelp_o_matic/models.py index a52d71d..78f3654 100644 --- 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":