app

app.py

@@ -73,24 +73,25 @@ def label_to_color_image(label):
 def draw_plot(pred_img, seg):
     fig = plt.figure(figsize=(20, 15))
 
-    grid_spec = gridspec.GridSpec(1, 2, width_ratios=[6, 1])
+    grid_spec = gridspec.GridSpec(1, 2, width_ratios=[1, 6])
 
-    plt.subplot(grid_spec[0])
-    plt.imshow(pred_img)
-    plt.axis('off')
+    plt.subplot(grid_spec[1])
     LABEL_NAMES = np.asarray(labels_list)
     FULL_LABEL_MAP = np.arange(len(LABEL_NAMES)).reshape(len(LABEL_NAMES), 1)
     FULL_COLOR_MAP = label_to_color_image(FULL_LABEL_MAP)
 
     unique_labels = np.unique(seg.numpy().astype("uint8"))
-    ax = plt.subplot(grid_spec[1])
-    plt.imshow(FULL_COLOR_MAP[unique_labels].astype(np.uint8), interpolation="nearest")
+    ax = plt.imshow(FULL_COLOR_MAP[unique_labels].astype(np.uint8), interpolation="nearest")
     ax.yaxis.tick_right()
     plt.yticks(range(len(unique_labels)), LABEL_NAMES[unique_labels])
     plt.xticks([], [])
     ax.tick_params(width=0.0, labelsize=25)
-    return fig
 
+    plt.subplot(grid_spec[0])
+    plt.imshow(pred_img)
+    plt.axis('off')
+
+    return fig
 def sepia(input_img):
     input_img = Image.fromarray(input_img)