I’m not the best at color matching, but I guess that the “yellow” output for channel5 is at ~0.16, while the output for channel0 would be at ~0.2. To better compare these outputs you could use the same scale for the probabilities (use [0, 1] as it would be the limit) or compare the values directly.