Refactor image reconstruction interface to remove unused code and improve layout

app.py

@@ -74,17 +74,17 @@ def visualize_single_image_no_mode(image_path):
     # make the plt figure larger
     plt.figure(figsize=(18, 8))
 
-    plt.subplot(1, 4, 1)
+    plt.subplot(1, 3, 1)
     show_image(img, "original")
 
-    plt.subplot(1, 4, 2)
+    plt.subplot(1, 3, 2)
     show_image(im_masked, "masked")
 
-    plt.subplot(1, 4, 3)
-    show_image(predicted_img, "reconstruction")
+    # plt.subplot(1, 4, 3)
+    # show_image(predicted_img, "reconstruction")
 
-    plt.subplot(1, 4, 4)
-    show_image(im_paste, "reconstruction + visible")
+    plt.subplot(1, 3, 3)
+    show_image(im_paste, "reconstruction")
 
     plt.tight_layout()
 
@@ -116,17 +116,17 @@ def visualize_single_image_pca_mode_bottom(image_path):
     # make the plt figure larger
     plt.figure(figsize=(18, 8))
 
-    plt.subplot(1, 4, 1)
+    plt.subplot(1, 3, 1)
     show_image(img, "original")
 
-    plt.subplot(1, 4, 2)
+    plt.subplot(1, 3, 2)
     show_image(im_masked, "masked")
 
-    plt.subplot(1, 4, 3)
+    plt.subplot(1, 3, 3)
     show_image(predicted_img, "reconstruction")
 
-    plt.subplot(1, 4, 4)
-    show_image(im_paste, "reconstruction + visible")
+    # plt.subplot(1, 4, 4)
+    # show_image(im_paste, "reconstruction + visible")
 
     plt.tight_layout()
 
@@ -158,17 +158,17 @@ def visualize_single_image_pca_mode_top(image_path):
     # make the plt figure larger
     plt.figure(figsize=(18, 8))
 
-    plt.subplot(1, 4, 1)
+    plt.subplot(1, 3, 1)
     show_image(img, "original")
 
-    plt.subplot(1, 4, 2)
+    plt.subplot(1, 3, 2)
     show_image(im_masked, "masked")
 
-    plt.subplot(1, 4, 3)
+    plt.subplot(1, 3, 3)
     show_image(predicted_img, "reconstruction")
 
-    plt.subplot(1, 4, 4)
-    show_image(im_paste, "reconstruction + visible")
+    # plt.subplot(1, 4, 4)
+    # show_image(im_paste, "reconstruction + visible")
 
     plt.tight_layout()