From 913b3cf77cc247315d7f86bd81e51936cc9f32da Mon Sep 17 00:00:00 2001
From: Aadi Redhu <90816714+AndyWarrior123@users.noreply.github.com>
Date: Tue, 30 Jul 2024 18:58:36 +0530
Subject: [PATCH 1/2] Update PredictionWithAugmentations.ipynb

Made the change to swap axes
---
 examples/PredictionWithAugmentations.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/PredictionWithAugmentations.ipynb b/examples/PredictionWithAugmentations.ipynb
index 5a2e88c..e5a66f3 100644
--- a/examples/PredictionWithAugmentations.ipynb
+++ b/examples/PredictionWithAugmentations.ipynb
@@ -216,7 +216,7 @@
     "            prediction = prediction.sigmoid()\n",
     "            \n",
     "            prediction_binary = convert_tensor_to_numpy(prediction.swapaxes(-1, 1))\n",
-    "            transformation.restore_fragment(prediction_binary)\n",
+    "            transformation.restore_fragment(prediction_binary.swapaxes(1,2))\n",
     "\n",
     "tta.merge()"
    ]

From 5c64033ae501536c4a0f2d694c9d9be736bc7f08 Mon Sep 17 00:00:00 2001
From: Aadi Redhu <90816714+AndyWarrior123@users.noreply.github.com>
Date: Tue, 30 Jul 2024 19:07:53 +0530
Subject: [PATCH 2/2] Update requirements.txt

removed numpy version requirement from the requirements.txt
---
 requirements.txt | 1 -
 1 file changed, 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index cb0f61f..da47b1e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,5 @@
 PyYAML == 5.3.1
 albumentations == 1.3.0
-numpy == 1.19.1
 opencv_python_headless == 4.4.0.44
 py_oneliner == 0.0.1
 scikit_learn == 0.23.2