tensorflow - 使用 Google 的 DEEPLAB V3+ 获取图像分割中每个语义类的类别概率分数

Question

在使用 Google 的 DeepLab V3+ 执行语义分割时，我试图为图像中存在的每个语义类提取像素级概率。我正在关注这里给出的演示 - https://github.com/tensorflow/models/tree/master/research/deeplab 。

我有以下问题 - 1. 如何使用 Google 的 DEEPLAB V3+ 获得图像分割中每个语义类的类别概率分数。2. 每个班级的门槛是多少？

将不胜感激这方面的任何帮助。

谢谢 ：）

下面给出的链接有我正在使用的网络 DeepLab V3+ 的图形定义 - https://paste.fedoraproject.org/paste/pbDVRLOaUgtZmT-rbEVmbQ

Answer 1

您可以使用以下方法获得概率： Deeplab 代码

predictions[output + PROB_SUFFIX] = tf.nn.softmax(logits)

编辑：最新的 deeplab 现在也提供概率。