small fixes (#200)

- string comparison for lower-contrast in DefectAugment - remove repetitive assignment in SimpleAugment
funkelab · Jan 22, 2024 · 92e2756 · 92e2756
1 parent 3686130
commit 92e2756
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/gunpowder/nodes/defect_augment.py b/gunpowder/nodes/defect_augment.py
@@ -190,7 +190,7 @@ def process(self, batch, request):
             if augmentation_type == "zero_out":
                 raw.data[section_selector] = 0
 
-            elif augmentation_type == "low_contrast":
+            elif augmentation_type == "lower_contrast":
                 section = raw.data[section_selector]
 
                 mean = section.mean()

diff --git a/gunpowder/nodes/simple_augment.py b/gunpowder/nodes/simple_augment.py
@@ -176,7 +176,7 @@ def process(self, batch, request):
             array.data = array.data[channel_slices + mirror]
 
             transpose = [t + num_channels for t in self.transpose]
-            array.data = array.data = array.data.transpose(
+            array.data = array.data.transpose(
                 list(range(num_channels)) + transpose
             )