search

rishizek / tensorflow-deeplab-v3

DeepLabv3 built in TensorFlow
MIT License
286 stars 102 forks source link

why do not apply softmax before predicting class? #34

Closed MaybeShewill-CV closed 5 years ago

MaybeShewill-CV commented 5 years ago

HI I wonder why don't you apply softmax on the inference logits before you decode the predicted class in file https://github.com/rishizek/tensorflow-deeplab-v3/blob/master/deeplab_model.py line 159~161. Thanks for your reply

MaybeShewill-CV commented 5 years ago

@rishizek Thanks I have solved this problem:)