Skip to content

Commit a740089

Browse files
committed
Revert "fix None value use for channelwise normalization."
This reverts commit 79b8df2.
1 parent a281d96 commit a740089

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pytorch3dunet/augment/transforms.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -567,22 +567,22 @@ def __call__(self, m):
567567
# get min/max channelwise
568568
axes = list(range(m.ndim))
569569
axes = tuple(axes[1:])
570-
if self.min_value is None:
570+
if self.min_value is None or 'None' in self.min_value:
571571
min_value = np.min(m, axis=axes, keepdims=True)
572572

573-
if self.max_value is None:
573+
if self.max_value is None or 'None' in self.max_value:
574574
max_value = np.max(m, axis=axes, keepdims=True)
575575

576576
# check if non None in self.min_value/self.max_value
577577
# if present and if so copy value to min_value
578578
if self.min_value is not None:
579579
for i,v in enumerate(self.min_value):
580-
if v is not None:
580+
if v != 'None':
581581
min_value[i] = v
582582

583583
if self.max_value is not None:
584584
for i,v in enumerate(self.max_value):
585-
if v is not None:
585+
if v != 'None':
586586
max_value[i] = v
587587
else:
588588
if self.min_value is None:

0 commit comments

Comments
 (0)