Fixed normalizeMinMax in U-Nets

HenriquesLab · Aug 1, 2023 · 021c2c0 · 021c2c0
1 parent 8b0f1e0
commit 021c2c0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Colab_notebooks/U-Net_2D_Multilabel_ZeroCostDL4Mic.ipynb b/Colab_notebooks/U-Net_2D_Multilabel_ZeroCostDL4Mic.ipynb
@@ -611,7 +611,7 @@
         "# Simple normalization to min/max fir the Mask\n",
         "def normalizeMinMax(x, dtype=np.float32):\n",
         "  x = x.astype(dtype,copy=False)\n",
-        "  x = (x - np.amin(x)) / (np.amax(x) - np.amin(x))\n",
+        "  x = (x - np.amin(x)) / (np.amax(x) - np.amin(x) + 1e-10)\n",
         "  return x\n",
         "\n",
         "\n",

diff --git a/Colab_notebooks/U-Net_2D_ZeroCostDL4Mic.ipynb b/Colab_notebooks/U-Net_2D_ZeroCostDL4Mic.ipynb
@@ -585,7 +585,7 @@
         "# Simple normalization to min/max fir the Mask\n",
         "def normalizeMinMax(x, dtype=np.float32):\n",
         "  x = x.astype(dtype,copy=False)\n",
-        "  x = (x - np.amin(x)) / (np.amax(x) - np.amin(x))\n",
+        "  x = (x - np.amin(x)) / (np.amax(x) - np.amin(x) + 1e-10)\n",
         "  return x\n",
         "\n",
         "\n",