rename TRAILMAP_test --> _MS

Author: MMathisLab

napari_cellseg3d/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.1"
+__version__ = "0.0.3"

napari_cellseg3d/model_framework.py

@@ -16,7 +16,7 @@
 from napari_cellseg3d.models import model_SegResNet as SegResNet
 from napari_cellseg3d.models import model_TRAILMAP as TRAILMAP
 from napari_cellseg3d.models import model_VNet as VNet
-from napari_cellseg3d.models import TRAILMAP_test as TMAP
+from napari_cellseg3d.models import TRAILMAP_MS as TMAP
 from napari_cellseg3d.plugin_base import BasePluginFolder
 
 warnings.formatwarning = utils.format_Warning
@@ -68,7 +68,7 @@ def __init__(self, viewer: "napari.viewer.Viewer"):
         }
         """dict: dictionary of available models, with string for widget display as key
 
-        Currently implemented : SegResNet, VNet, TRAILMAP_test"""
+        Currently implemented : SegResNet, VNet, TRAILMAP_MS"""
 
         self.worker = None
         """Worker from model_workers.py, either inference or training"""

napari_cellseg3d/models/TRAILMAP_test.py napari_cellseg3d/models/TRAILMAP_MS.pynapari_cellseg3d/models/TRAILMAP_test.py renamed to napari_cellseg3d/models/TRAILMAP_MS.py

@@ -8,7 +8,7 @@ def get_weights_file():
 
 
 def get_net():
-    return TRAILMAP_test(1, 1)
+    return TRAILMAP_MS(1, 1)
 
 
 def get_output(model, input):
@@ -22,7 +22,7 @@ def get_validation(model, val_inputs):
     return model(val_inputs)
 
 
-class TRAILMAP_test(nn.Module):
+class TRAILMAP_MS(nn.Module):
     def __init__(self, in_ch, out_ch):
         super().__init__()
         self.conv0 = self.encoderBlock(in_ch, 32, 3)  # input