时间戳

首页 > 解决方案 > 使用 Tensorflow 的 DeepLab 输出黑屏

python - 使用 Tensorflow 的 DeepLab 输出黑屏

问题描述

我在我自己的数据集上训练了 DeepLab，initialize_last_layer并且last_layers contain_logits只使用了两者False。我只有 2 类背景和标签。我尝试使用--num_classes=2但3没有任何效果。我的ignore_label 是255，而我的背景是(0,0,0,255)，标签是(1,1,1,255)。我也给了label_weights 1:20但没用。我看到的只是segmented_results文件夹中的黑色图像。有什么解决办法吗？

标签： pythontensorflowmodeltensorflow2.0deeplab

解决方案

在您的vis.sh

例如。：

--colormap_type="cityscapes"

推荐阅读