Update layout-fine-tune.ipynb

layout-fine-tune.ipynb

@@ -69,7 +69,7 @@
     "os.makedirs(checkpoint_dir, exist_ok=True)\n",
     "writer = SummaryWriter(log_dir=log_dir)\n",
     "\n",
-    "def logits_to_bboxes(logits,image) : # This function is useful for converting the mask into bounding boxes.(The model does not provide bounding boxes.)\n",
+    "def logits_to_bboxes(logits,image) : # This function is useful for converting the logits(mask) into bounding boxes.(The model does not provide bounding boxes.)\n",
     "    correct_shape = (300, 300)  \n",
     "    logits_temp = F.interpolate(logits, size=correct_shape, mode='bilinear', align_corners=False)\n",
     "    logits_temp = logits_temp.cpu().detach().numpy().astype(np.float32)\n",