Fix scale computation in attention visualization example (#210)

facebookresearch · Aug 21, 2020 · 3e92cd7 · 3e92cd7
1 parent f6bc5d2
commit 3e92cd7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebooks/detr_attention.ipynb b/notebooks/detr_attention.ipynb
@@ -1302,7 +1302,7 @@
         "fcenter_ax = fig.add_subplot(gs[:, 1:-1])\n",
         "fcenter_ax.imshow(im)\n",
         "for (y, x) in idxs:\n",
-        "    scale = im.height / img.shape[-1]\n",
+        "    scale = im.height / img.shape[-2]\n",
         "    x = ((x // fact) + 0.5) * fact\n",
         "    y = ((y // fact) + 0.5) * fact\n",
         "    fcenter_ax.add_patch(plt.Circle((x * scale, y * scale), fact // 2, color='r'))\n",
@@ -1619,4 +1619,4 @@
       ]
     }
   ]
-}
+}