【深度学习框架输入格式】NCHW还是NHWC?
文章目录1 三大框架使用格式2 NCHW和NHWC优缺点对比3 如何从输入图像到想要的格式呢?1 三大框架使用格式PyTorch:NCHWCaffe:NCHWTensorFlow:默认使用NHWC,GPU也支持NCHW2 NCHW和NHWC优缺点对比NCHW 必须等所有通道输入准备好才能得到最终输出结果,需要占用较大的临时空间。NHWC 的访存局部性更好(每三个输入像素即可得到一个输出像素);Te
·
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小节。
更多推荐
所有评论(0)