That is strange.
But about probabilities. Could you please explain how probabilities are not high enough? I mean, what is your comparison metric or let’s say for a image x, what do you expect and what are probs currently you obtaining?