How did you load your images?
As far as I know, numpy doesn’t support image loading out of the box, instead other libraries like scipy support it.
What happens if you just swap your axes? Does the model still yield bad results?
I understand that after downgrading to the previous setup, you can’t reproduce the original results?