Las cuatro convenciones de ejes
Lee cada convención como una frase:
HWC— alto × ancho × colorCHW— color × alto × anchoNHWC— ejemplos × alto × ancho × colorNCHW— ejemplos × color × alto × ancho
scikit-image y la mayoría de los archivos de imagen te dan HWC. PyTorch quiere NCHW. TensorFlow quiere NHWC. Ninguna es correcta ni incorrecta; simplemente no coinciden, y tienes que saber cuál tienes en la mano.