quick fix for ipynb.checkpoints error

shreyas-bk · Oct 29, 2020 · f059e69 · f059e69
1 parent 7cfeb08
commit f059e69
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/DEMOS/U_2_Netp_Demonstration_Colab.ipynb b/DEMOS/U_2_Netp_Demonstration_Colab.ipynb
@@ -252,6 +252,8 @@
         "\n",
         "  # BACKGROUND REMOVAL\n",
         "\n",
+        "  if name == '.ipynb_checkpo'\n:",
+        "    continue\n",
         "  output = load_img('/content/U-2-Net/results/'+name+'.png')\n",
         "  out_img = img_to_array(output)\n",
         "  out_img /= RESCALE\n",