使用Tensorflow的DeepLab输出黑屏

Question

我在我自己的数据集上训练了DeepLab，只使用initialize_last_layer和last_layers contain_logits，都使用False。我只有两个类别的背景和标签。我尝试使用--num_classes=2和3，但都不起作用。我的ignore_label是255，而我的背景是(0,0,0,255)，label是(1,1,1,255)。我也给了label_weights 1:20，但没有用。我在segmented_results文件夹中看到的都是黑色图像。有什么解决方案吗？

Answer 1

在vis.sh中定义一些色彩映射表

例如：

--colormap_type="cityscapes"