app

app.py

@@ -71,12 +71,11 @@ def label_to_color_image(label):
     return colormap[label]
 
 def draw_plot(pred_img, seg):
-    fig = plt.figure(figsize=(30, 15))  # figsize를 조절하여 전체 그림 크기 변경
+    fig = plt.figure(figsize=(20, 15))
 
-    # 그리드의 구성을 변경하여 입력 상자와 출력 상자의 위치를 바꿈
-    grid_spec = gridspec.GridSpec(1, 2, width_ratios=[2, 8])  # width_ratios 값을 조절하여 비율 변경
 
-    # 출력 상자를 먼저 플로팅
+    grid_spec = gridspec.GridSpec(1, 2, width_ratios=[1, 6])
+
     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)
@@ -89,7 +88,6 @@ def draw_plot(pred_img, seg):
     plt.xticks([], [])
     ax.tick_params(width=0.0, labelsize=25)
 
-    # 입력 상자를 다음으로 플로팅
     ax = plt.subplot(grid_spec[0])
     ax.imshow(pred_img)
     ax.axis('off')