ValueError：有 5 个颜色箱（包括扩展），但 ncolors = 4； ncolors 必须等于或大于 bin 的数量

Question

我正在栅格化我的 Shapefile，以便使用以下代码使用 U-Net 创建语义分割标签：

rstzr = r.Rasterizer(shape_file,
                     raster_file,
                     class_column,
                     classes_interest = classes_of_interest,
                     non_class_name = non_class)

rstzr.collect_class_names()
rstzr.rasterize_layer()
rstzr.remove_labels_under_cloud(pos_qa = 0, new_label = 3)
m_class_names = ['No Data'] + rstzr.get_class_names() + ['Clouds']
rasterized_layer = rstzr.get_labeled_raster()

rstzr.save_labeled_raster_to_gtiff(out_labels)

vis.plot_labels(rasterized_layer, m_class_names, colors = ['white', 'green', 'yellow', 'blue'], title = "Rasterized Ground Truth")

当我在 Google Colab 中使用此代码时 光栅化形状文件，它工作正常，并且光栅化地面真相也正在绘制，如屏幕截图所示。但是，当我在我大学的其他具有高计算性能的 Jupyter 平台上尝试相同的代码时，它显示错误，可以在屏幕截图中看到

任何人都可以建议该错误可能是什么问题

ValueError：有 5 个颜色箱（包括扩展），但 ncolors = 4； ncolors 必须等于或超过 bin 数量

才能正常工作？