Per-channel normalization option in segmentation preprocessing.

BodenmillerGroup · Jun 4, 2021 · 30df6e6 · 30df6e6
1 parent 97b8009
commit 30df6e6
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 4 deletions.
diff --git a/backend/histocat/core/segmentation/dto.py b/backend/histocat/core/segmentation/dto.py
@@ -6,6 +6,7 @@
 class SegmentationPreprocessingSettingsDto(BaseModel):
     """Segmentation preprocessing settings model."""
 
+    channels_normalization: Optional[str]
     threshold: Optional[bool]
     percentile: Optional[float]
     normalize: bool

diff --git a/backend/histocat/worker/segmentation/processors/acquisition_processor.py b/backend/histocat/worker/segmentation/processors/acquisition_processor.py
@@ -41,11 +41,16 @@ def process_acquisition(
     acquisition_data = parser.get_acquisition_data()
 
     nuclei_channels = acquisition_data.get_image_stack_by_names(params.nuclei_channels)
-    # nuclei_channels = normalize_by_zscore(nuclei_channels)
-    nuclei_channels = np.nanmean(nuclei_channels, axis=0)
-
     cytoplasm_channels = acquisition_data.get_image_stack_by_names(params.cytoplasm_channels)
-    # cytoplasm_channels = normalize_by_zscore(cytoplasm_channels)
+
+    if params.preprocessing.channels_normalization == "minmax":
+        nuclei_channels = normalize_by_minmax(nuclei_channels)
+        cytoplasm_channels = normalize_by_minmax(cytoplasm_channels)
+    elif params.preprocessing.channels_normalization == "zscore":
+        nuclei_channels = normalize_by_zscore(nuclei_channels)
+        cytoplasm_channels = normalize_by_zscore(cytoplasm_channels)
+
+    nuclei_channels = np.nanmean(nuclei_channels, axis=0)
     cytoplasm_channels = np.nanmean(cytoplasm_channels, axis=0)
 
     # Combined together and expand to 4D

diff --git a/frontend/src/modules/segmentation/models.ts b/frontend/src/modules/segmentation/models.ts
@@ -1,4 +1,5 @@
 export interface ISegmentationPreprocessingSettings {
+  channels_normalization?: string;
   threshold?: boolean;
   percentile?: number;
   normalize: boolean;

diff --git a/frontend/src/views/group/project/segmentation/SegmentationSettingsView.vue b/frontend/src/views/group/project/segmentation/SegmentationSettingsView.vue
@@ -23,6 +23,11 @@
           <v-text-field label="Dataset Description" hint="Resulting dataset description" v-model="datasetDescription" />
         </v-tab-item>
         <v-tab-item>
+          <v-radio-group v-model="channelsNormalization" label="Per-channel normalization" dense>
+            <v-radio label="None" value="none" />
+            <v-radio label="Min-Max" value="minmax" />
+            <v-radio label="Z-score" value="zscore" />
+          </v-radio-group>
           <v-switch v-model="threshold" label="Threshold" inset class="ml-1" dense />
           <v-text-field
             label="Percentile"
@@ -173,6 +178,7 @@ export default class SegmentationSettingsView extends Vue {
   modelId: number | null = null;
   compartment = "whole-cell";
 
+  channelsNormalization = "none";
   threshold = true;
   percentile = 99.9;
   normalize = true;
@@ -214,6 +220,7 @@ export default class SegmentationSettingsView extends Vue {
       }
 
       const preprocessingParams = {
+        channels_normalization: this.channelsNormalization,
         threshold: this.threshold,
         percentile: this.percentile,
         normalize: this.normalize,