Merge pull request #6 from janelia-cellmap/load_checkpoints

Cosem classes channels
janelia-cellmap · Mar 19, 2024 · 9f36dde · 9f36dde
2 parents 50a9230 + 191ce31
commit 9f36dde
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 0 deletions.
diff --git a/src/cellmap_models/pytorch/cosem/setup04/model.py b/src/cellmap_models/pytorch/cosem/setup04/model.py
@@ -24,3 +24,19 @@
 final_feature_width = 12 * 6
 
 classes_out = 14
+classes_channels = [
+    "ecs",
+    "pm",
+    "mito",
+    "mito_mem",
+    "ves",
+    "ves_mem",
+    "endo",
+    "endo_mem",
+    "er",
+    "er_mem",
+    "eres",
+    "nuc",
+    "mt",
+    "mt_out",
+]
diff --git a/src/cellmap_models/pytorch/cosem/setup26_1/model.py b/src/cellmap_models/pytorch/cosem/setup26_1/model.py
@@ -24,3 +24,4 @@
 final_feature_width = 12 * 6
 
 classes_out = 3
+classes_channels = ["mito", "mito_mem", "mito_ribo"]
diff --git a/src/cellmap_models/pytorch/cosem/setup28/model.py b/src/cellmap_models/pytorch/cosem/setup28/model.py
@@ -24,3 +24,4 @@
 final_feature_width = 12 * 6
 
 classes_out = 2
+classes_channels = ["er", "er_mem"]
diff --git a/src/cellmap_models/pytorch/cosem/setup36/model.py b/src/cellmap_models/pytorch/cosem/setup36/model.py
@@ -24,3 +24,4 @@
 final_feature_width = 12 * 6
 
 classes_out = 2
+classes_channels = ["nuc", "nucleo"]
diff --git a/src/cellmap_models/pytorch/cosem/setup45/model.py b/src/cellmap_models/pytorch/cosem/setup45/model.py
@@ -24,3 +24,4 @@
 final_feature_width = 12 * 6
 
 classes_out = 2
+classes_channels = ["ecs", "pm"]