Inference models may skip the final softmax op of the network. This means the probabilities printed by the ImageClassifier are incorrect, as it expects softmax to be the final op. We could calculate the softmax in the ImageClassifier itself. This would allow the user to see the probabilities even if the network is missing it. This would likely make sense as a command line option.
For testing purposes, you could use an existing model from Glow model zoo: densenet121
To have the problem description self contained, here is the semantic of softmax:
res_i = exp(input_i) / (exp(input_0) + ... + exp(input_N)).
Most helpful comment
To have the problem description self contained, here is the semantic of softmax:
res_i = exp(input_i) / (exp(input_0) + ... + exp(input_N)).