1 三大框架使用格式
PyTorch:NCHW
Caffe:NCHW
TensorFlow:默認(rèn)使用NHWC叫确,GPU也支持NCHW
2 NCHW和NHWC優(yōu)缺點對比
NCHW 必須等所有通道輸入準(zhǔn)備好才能得到最終輸出結(jié)果归露,需要占用較大的臨時空間熙卡。
NHWC 的訪存局部性更好(每三個輸入像素即可得到一個輸出像素)杖刷;
TensorFlow 為什么選擇 NHWC 格式作為默認(rèn)格式?
因為早期開發(fā)都是基于 CPU的驳癌,使用 NHWC 比 NCHW 稍快一些滑燃。
NCHW 則是 Nvidia cuDNN 默認(rèn)格式,使用 GPU 加速時用 NCHW 格式速度會更快颓鲜。
3 如何從輸入圖像到想要的格式呢表窘?
見參考鏈接:
http://www.reibang.com/p/5b53af742ad5