В функции
def predict_labels(images, model_options, image_pyramid=None)
из model.py, у нас есть:
predictions[output] = tf.argmax(logits, 3)
https://github.com/tensorflow/models/blob/1c99681ed94495f446399e247a8d63e07f28391c/research/deeplab/model.py#L191
Я ожидал что-то вроде:
predictions[output] = tf.argmax(tf.nn.softmax(logits), 3)
Так почему там нет софтмакса?