数据通道：NCHW和NHWC

文章来自微信公众号“科文路”，欢迎关注、互动。转发须注明出处。

考虑到数据传输带宽和计算性能，在深度学习中，图像数据通道格式主要有两种：NCHW和NHWC

基本概念

• N - Batch
• C - Channel
• H - Height
• W - Width

NCHW

想象成 4 维数组

先取 W 方向，然后 H、C、N

NHWC

想象成 4 维数组

先取 C 方向，然后 W、H、N

一图释义

一些结论

• 在 GPU 中使用 NCHW 格式计算卷积更快
• 在 CPU 中使用 NHWC 格式，有SSE 或 AVX优化，计算卷积更快

应用

• TensorFlow：默认 NHWC，GPU 支持 NCHW
• Caffe：NCHW
• PyTorch：NCHW

参考

图像数据通道格式：NCHW和NHWC的区别

图解NCHW与NHWC数据格式

都看到这儿了，不如关注每日推送的“科文路”、互动起来~