CHW,HCW顺序和numpy.transpose函数

改Aligned_ReID代码的时候碰到输入是CHW 顺序还是HCW顺序的问题,(Caffe Prefers CHW Order,quote from caffe tutorial)

more see on https://caffe2.ai/docs/tutorial-image-pre-processing.html

改image pre_process考虑改换CHW顺序

其中改换顺序需要用到的是numpy.transpose()函数,为了读懂这个函数,以下这个比较生动

https://blog.csdn.net/Hearthougan/article/details/72626643?locationNum=7&fps=1

numpy.transpose()

虽然上面的transpose讲的挺详细的,但是还是不是很清楚transpose后跟之前相比的变换,

于是下面自己做了一个实验,第一列是结构,第二列是做的变换

[3, 6, 256, 128]     原来的

[256, 6, 3, 128]    transpose(0, 2, 1)

[256, 6, 128, 3]    transpose(0 , 1, 2)

[256, 6, 128, 3]     transpose(2, 0, 1)

[3, 6, 128, 256]     transpose(2, 1, 0)

[128, 6, 256, 3]    transpose(1, 0, 2)

[128, 6, 3, 256]     transpose(1, 2, 0)

...第二个的6雷打不动...我....

猜你喜欢

转载自blog.csdn.net/ptgood/article/details/82949912
今日推荐