转换图像的通道位置的时候会用到
例如:
对于256*256的三通道彩色图像,cv2以及PIL.Image等读出的numpy格式图像,通道数都在最后(256,256,3)
而有时候会需要channel first,(3,256,256)
就可以使用
np.transpose(img,(2,0,1))
相当于从(0,1,2)到(2,0,1)
转换图像的通道位置的时候会用到
例如:
对于256*256的三通道彩色图像,cv2以及PIL.Image等读出的numpy格式图像,通道数都在最后(256,256,3)
而有时候会需要channel first,(3,256,256)
就可以使用
np.transpose(img,(2,0,1))
相当于从(0,1,2)到(2,0,1)