【深度学习框架输入格式】NCHW还是NHWC？

1 三大框架使用格式

PyTorch：NCHW
Caffe：NCHW
TensorFlow：默认使用NHWC，GPU也支持NCHW

2 NCHW和NHWC优缺点对比

NCHW 必须等所有通道输入准备好才能得到最终输出结果，需要占用较大的临时空间。

NHWC 的访存局部性更好（每三个输入像素即可得到一个输出像素）；

TensorFlow 为什么选择 NHWC 格式作为默认格式？
因为早期开发都是基于 CPU的，使用 NHWC 比 NCHW 稍快一些。

NCHW 是 Nvidia cuDNN 默认格式，使用 GPU 加速时用 NCHW 格式速度会更快。

3 如何从输入图像到想要的格式呢？

见参考链接 PIL和opencv读取、显示图片+归一化+transpose变换通道 中第3小节。