From 6ca1a3de94c91661564b99f77cd806628079b0a5 Mon Sep 17 00:00:00 2001 From: pranavbrkr Date: Wed, 4 Oct 2023 15:34:27 -0700 Subject: [PATCH] more comments --- Phase 2/task1.ipynb | 3 +++ 1 file changed, 3 insertions(+) diff --git a/Phase 2/task1.ipynb b/Phase 2/task1.ipynb index e1063a0..90c6840 100644 --- a/Phase 2/task1.ipynb +++ b/Phase 2/task1.ipynb @@ -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",