Получите оценки вероятности для каждого семантического класса в сегментации изображений с помощью Google DEEPLAB V3 +

Я пытаюсь извлечь пиксельные вероятности для каждого семантического класса, присутствующего в изображении, при выполнении семантической сегментации с использованием 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


tensorflow image-segmentation deeplab
person Sid Gairola    schedule 03.07.2019    source источник

Ответы (1)


arrow_upward
2
arrow_downward

Вы можете получить вероятность, используя следующее: Код Deeplab

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

РЕДАКТИРОВАТЬ: последняя версия deeplab теперь также предоставляет вероятности.

person Neeraj Gulia    schedule 19.11.2019