Default params

Update the default parameters for the original implementation.

Author: sauravmishra1710

UNet - Biomedical_Segmentation.ipynb

@@ -84,15 +84,22 @@
     "# Input shape - (572, 572, 1), \n",
     "# filters 64, 128, 256, 512, 1024 at each convolutional block and \n",
     "# unpadded convolutions.\n",
-    "unet = Unet(input_shape = (572, 572, 1), filters = [64, 128, 256, 512, 1024], padding = \"valid\")\n",
-    "\n",
+    "unet = Unet()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "# call the build netowrk API to build the network.\n",
     "model = unet.Build_UNetwork()"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [
     {
@@ -194,7 +201,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 5,
    "metadata": {
     "scrolled": false
    },

Unet.py

@@ -21,7 +21,7 @@ class Unet():
         https://arxiv.org/pdf/1505.04597
     """
 
-    def __init__(self, input_shape, filters, padding):
+    def __init__(self, input_shape = (572, 572, 1), filters = [64, 128, 256, 512, 1024], padding = "valid"):
         """
         
         Initialize the Unet framework and the model parameters - input_shape,