【深度学习框架输入格式】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小节。

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值