diff --git a/src/cellmap_models/pytorch/cosem/setup04/model.py b/src/cellmap_models/pytorch/cosem/setup04/model.py
index 437f85f..613e0f5 100755
--- 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
index bc9916e..f13ce4a 100755
--- 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
index a3d4e36..2689c3d 100755
--- 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
index a3d4e36..2abfad9 100755
--- 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
index 938344e..2b4d995 100755
--- 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"]