remove torch requirement

examples/python_unet/unet_inference.ipynb

@@ -18,8 +18,7 @@
     "import migraphx\n",
     "from PIL import Image\n",
     "import numpy as np\n",
-    "import matplotlib.pyplot as plt\n",
-    "import torch"
+    "import matplotlib.pyplot as plt"
    ]
   },
   {
@@ -159,6 +158,17 @@
     "output_mask.shape"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "acbd68e3",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def sigmoid(x):\n",
+    "  return 1 / (1 + np.exp(-x))"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -166,7 +176,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "probs = torch.sigmoid(torch.from_numpy(output_mask))\n",
+    "probs = sigmoid(output_mask)\n",
     "full_mask = probs > 0.996\n",
     "plot_img_and_mask(imPrint, full_mask)"
    ]