Fix output of probs image in chosen format.