app
Browse files
app.py
CHANGED
@@ -71,10 +71,10 @@ def label_to_color_image(label):
|
|
71 |
return colormap[label]
|
72 |
|
73 |
def draw_plot(pred_img, seg):
|
74 |
-
fig = plt.figure(figsize=(
|
75 |
|
76 |
|
77 |
-
grid_spec = gridspec.GridSpec(1, 2, width_ratios=[
|
78 |
|
79 |
ax = plt.subplot(grid_spec[1])
|
80 |
LABEL_NAMES = np.asarray(labels_list)
|
|
|
71 |
return colormap[label]
|
72 |
|
73 |
def draw_plot(pred_img, seg):
|
74 |
+
fig = plt.figure(figsize=(30, 15))
|
75 |
|
76 |
|
77 |
+
grid_spec = gridspec.GridSpec(1, 2, width_ratios=[6, 1])
|
78 |
|
79 |
ax = plt.subplot(grid_spec[1])
|
80 |
LABEL_NAMES = np.asarray(labels_list)
|