a

app.py

@@ -81,7 +81,7 @@ def draw_plot(pred_img, seg):
 
     # Limit unique_labels to be within the range of colormap
     unique_labels = np.unique(seg.astype("uint8"))
-    unique_labels = unique_labels[unique_labels < len(colormap)]
+    unique_labels = unique_labels[unique_labels < len(FULL_COLOR_MAP)]
 
     ax = plt.subplot(grid_spec[1])
 
@@ -98,7 +98,6 @@ def draw_plot(pred_img, seg):
     plt.xticks([], [])
     ax.tick_params(width=0.0, labelsize=25)
     return fig
-
 def label_to_color_image(label):
     if label.ndim != 2:
         raise ValueError("Expect 2-D input label")