Disable TRAILMAP

Removed draft of TRAILMAP port to Pytorch

Author: C-Achard

napari_cellseg3d/model_framework.py

@@ -14,7 +14,7 @@
 from napari_cellseg3d.log_utility import Log
 from napari_cellseg3d.models import model_SegResNet as SegResNet
 from napari_cellseg3d.models import model_SwinUNetR as SwinUNetR
-from napari_cellseg3d.models import model_TRAILMAP as TRAILMAP
+# from napari_cellseg3d.models import model_TRAILMAP as TRAILMAP
 from napari_cellseg3d.models import model_VNet as VNet
 from napari_cellseg3d.models import model_TRAILMAP_MS as TRAILMAP_MS
 from napari_cellseg3d.plugin_base import BasePluginFolder
@@ -63,7 +63,7 @@ def __init__(self, viewer: "napari.viewer.Viewer"):
         self.models_dict = {
             "VNet": VNet,
             "SegResNet": SegResNet,
-            "TRAILMAP": TRAILMAP,
+            # "TRAILMAP": TRAILMAP,
             "TRAILMAP_MS": TRAILMAP_MS,
             "SwinUNetR": SwinUNetR,
         }

napari_cellseg3d/models/model_TRAILMAP.py

@@ -92,7 +92,6 @@ def bridgeBlock(self, in_ch, out_ch, kernel_size, padding="same"):
             ),
             nn.BatchNorm3d(out_ch),
             nn.ReLU(),
-            # nn.ConvTranspose3d(out_ch, out_ch, kernel_size=2, stride=2),
         )
         return encode
 
@@ -117,6 +116,5 @@ def outBlock(self, in_ch, out_ch, kernel_size, padding="same"):
 
         out = nn.Sequential(
             nn.Conv3d(in_ch, out_ch, kernel_size=kernel_size, padding=padding),
-            # nn.BatchNorm3d(out_ch),
         )
         return out

napari_cellseg3d/models/pretrained/pretrained_model_urls.json

@@ -1,6 +1,5 @@
 {
   "TRAILMAP_MS": "http://deeplabcut.rowland.harvard.edu/cellseg3dmodels/TRAILMAP_MS.tar.gz",
-  "TRAILMAP": "http://deeplabcut.rowland.harvard.edu/cellseg3dmodels/TRAILMAP.tar.gz",
   "SegResNet": "http://deeplabcut.rowland.harvard.edu/cellseg3dmodels/SegResNet.tar.gz",
   "VNet": "http://deeplabcut.rowland.harvard.edu/cellseg3dmodels/VNet.tar.gz"
 }