app

app.py

@@ -73,23 +73,26 @@ 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=[1, 6])
 
-    plt.subplot(grid_spec[1])
+    # 출력 상자를 먼저 플로팅
+    ax = 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.imshow(FULL_COLOR_MAP[unique_labels].astype(np.uint8), interpolation="nearest")
+    ax.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)
 
-    plt.subplot(grid_spec[0])
-    plt.imshow(pred_img)
-    plt.axis('off')
+    # 입력 상자를 다음으로 플로팅
+    ax = plt.subplot(grid_spec[0])
+    ax.imshow(pred_img)
+    ax.axis('off')
 
     return fig
 def sepia(input_img):