Update 06.ipynb

Author: Programmer-RD-AI

.virtual_documents/06.ipynb

@@ -195,18 +195,28 @@ assert img_size % patch_size == 0, "Image Size must be divisible by patch size"
 num_patches
 
 
-img_size / patch_size, img_size // patch_size
-
+img_size = 224
+patch_size = 16
 
-fig, axs = plt.subplots(nrows=1, ncols=img_size // patch_size, sharex=True, sharey=True)
-for i, patch in enumerate(range(0, img_size, patch_size)):
-    axs[i].imshow(image_permuted[:patch_size, patch:patch+patch_size, :]);
-    axs[i].set_xlabel(i+1)
-    axs[i].set_xticks([])
-    axs[i].set_yticks([])
 
+fig, axs = plt.subplots(nrows=img_size // patch_size, ncols=img_size // patch_size, figsize=(num_patches, num_patches), sharex=True, sharey=True)
+# Loop through height and width of the image
+for i, patch_size in enumerate(range(0, img_size, 16)):
+    for j, patch_width in enumerate(range(0, img_size, 16)):
+        # Plot the permuted image on the different axes
+        axs[i, j].imshow(image_permuted[patch_height:patch_height+patch_size, patch_width:patch_width+patch_size,:])
+        axs[i, j].set_ylabel(i+1, rotation="horizontal", horizontalalignment="right", verticalalignment="center")
+        axs[i, j].set_xlabel(j+1)
+        axs[i, j].set_xticks([])
+        axs[i, j].set_yticks([])
+        axs[i, j].label_outer()
+
+# Set up a title for the plot
+fig.suptitle(f"{class_names[label]} -> Patchified", fontsize=14)
+plt.show()
 
 
+patch_size