深度学习框架卷积时的数据格式

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/liuweiyuxiang/article/details/84951363

numpy image: H x W x C
torch image: C X H X W
pytorch:[b,channel,h,w]
tensorflow:[b,h,w,channel]
caffe:

猜你喜欢

转载自blog.csdn.net/liuweiyuxiang/article/details/84951363