Add slice_axis to SlidingWindowInferer

Signed-off-by: Suraj <b.pai@maastrichtuniversity.nl>

Author: surajpaib

modules/2d_inference_3d_volume.ipynb

@@ -41,8 +41,8 @@
      "output_type": "stream",
      "text": [
       "Requirement already satisfied: monai in /home/suraj/miniconda3/envs/monai/lib/python3.7/site-packages (0.8.0)\n",
-      "Requirement already satisfied: torch>=1.6 in /home/suraj/miniconda3/envs/monai/lib/python3.7/site-packages (from monai) (1.10.0)\n",
       "Requirement already satisfied: numpy>=1.17 in /home/suraj/miniconda3/envs/monai/lib/python3.7/site-packages (from monai) (1.21.4)\n",
+      "Requirement already satisfied: torch>=1.6 in /home/suraj/miniconda3/envs/monai/lib/python3.7/site-packages (from monai) (1.10.0)\n",
       "Requirement already satisfied: typing-extensions in /home/suraj/miniconda3/envs/monai/lib/python3.7/site-packages (from torch>=1.6->monai) (4.0.1)\n"
      ]
     }
@@ -133,7 +133,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 3,
    "id": "85b15305",
    "metadata": {},
    "outputs": [
@@ -175,7 +175,7 @@
     "\n",
     "# Create an instance of YourSlidingWindowInferer with roi_size as the 64x256 (DxW) and sliding over H axis\n",
     "coronal_inferer = YourSlidingWindowInferer(roi_size=(64, 256),\n",
-    "                                   sw_batch_size=1, slice_axis=0, # Slice axis is added here\n",
+    "                                   sw_batch_size=1, slice_axis=1, # Slice axis is added here\n",
     "                                   cval=-1)\n",
     "\n",
     "output = coronal_inferer(input_volume, net)\n",
@@ -187,7 +187,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2c6c0028",
+   "id": "454c353f",
    "metadata": {},
    "outputs": [],
    "source": []