channel first和channel last之间的转换方法

Posted 2022-09-19 bug404

在pytorch中，一般图像信息都是channel first，即通道数在前。比如一张84 x 84 x 3的图片，传到pytorch的时候需要按照(3, 84, 84)的shape传进去。那么怎样将(84, 84, 3)的shape转换成(3, 84, 84)以及相互转换呢？

x = np.zeros((12, 12, 3))
x.shape
#yields: 
(12, 12, 3)

x = np.moveaxis(x, -1, 0)
x.shape
#yields: 
(3, 12, 12)

x = np.zeros((3, 12, 12))
x.shape
#yields: 
(3, 12, 12)

x = np.moveaxis(x, 0, -1)
x.shape
#yields: 
(12, 12, 3)

看例子，当然是借助np.moveaxis函数了。moveaxis的第二个参数的意思是原始索引的位置，最后一个参数代表目标的索引位置。

---

参考：

​​https://stackoverflow.com/questions/43829711/what-is-the-correct-way-to-change-image-channel-ordering-between-channels-first​​