more comments

This commit is contained in:
pranavbrkr 2023-10-04 15:34:27 -07:00
parent cfda9fb4cd
commit 6ca1a3de94

View File

@ -151,6 +151,9 @@
" fig, axes = plt.subplots(1, k, figsize=(15, 5))\n",
"\n",
" for i in range(k):\n",
" # caltechDataset[similar_images[i][\"_id\"]][1] because\n",
" # similar_images[i][\"_id\"] will provide me the image id\n",
" # [1] will be image pixel values since caltechDataset is in format (id, pixels, label)\n",
" axes[i].imshow(caltechDataset[similar_images[i][\"_id\"]][1].permute(1, 2, 0))\n",
" axes[i].set_title(f'id: {similar_images[i][\"_id\"]}')\n",
"\n",