使用Tensorflow的DeepLab的黑屏输出

Question

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

Answer 1

在vis.sh中定义一些颜色图

例如：

--colormap_type="cityscapes"