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 如何从输入图像到想要的格式呢?
见参考链接:
https://www.jianshu.com/p/5b53af742ad5