Update keypoint.ipynb (WongKinYiu#292)

* Update keypoint.ipynb add "with torch.no_grad():" in line 60. * Update keypoint.ipynb
Katsileros · Aug 6, 2022 · 4789bd1 · 4789bd1
1 parent 591c5b0
commit 4789bd1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tools/keypoint.ipynb b/tools/keypoint.ipynb
@@ -57,7 +57,8 @@
    "outputs": [],
    "source": [
     "output = non_max_suppression_kpt(output, 0.25, 0.65, nc=model.yaml['nc'], nkpt=model.yaml['nkpt'], kpt_label=True)\n",
-    "output = output_to_keypoint(output)\n",
+    "with torch.no_grad():\n",
+    "    output = output_to_keypoint(output)\n",
     "nimg = image[0].permute(1, 2, 0) * 255\n",
     "nimg = nimg.cpu().numpy().astype(np.uint8)\n",
     "nimg = cv2.cvtColor(nimg, cv2.COLOR_RGB2BGR)\n",