Based on the error message numpy isn’t able to transpose the numpy array, which might happen e.g. if you are now loading grayscale images (and were loading RGB images previously), so you would have to make sure the same image format is used in both setups.