使用谷歌的DEEPLAB V3+获得图像分割中每个语义类的类概率分数

Question

当使用谷歌的DeepLab V3+进行语义分割时，我正在尝试提取图像中存在的每个语义类别的像素概率。我正在遵循这里给出的演示- https://github.com/tensorflow/models/tree/master/research/deeplab。

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

会非常感谢在这个方向上的任何帮助。

谢谢您:)

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

Answer 1

您可以使用以下命令来获得概率：Deeplab code

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

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