Define constants and configurations

Author: AAdewunmi

notebooks/04_bias_analysis.ipynb

@@ -46,6 +46,38 @@
     "\n",
     "from src.gradcam import GradCAM\n"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "id": "acd24d80",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Define constants and configurations\n",
+    "\n",
+    "DATA_DIR = Path(\"static/output\")   \n",
+    "MODEL_PATH = Path(\"saved_models/resnet50_best.pt\")\n",
+    "\n",
+    "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
+    "\n",
+    "transform = transforms.Compose([\n",
+    "    transforms.Resize((224,224)),\n",
+    "    transforms.ToTensor(),\n",
+    "    transforms.Normalize(\n",
+    "        mean=[0.485, 0.456, 0.406],\n",
+    "        std=[0.229, 0.224, 0.225]\n",
+    "    )\n",
+    "])\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "98b31098",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {