图像通道转换[n c h w] 转 [n h w c]

Tensorflow定义的tensor的shape为[n,h,w,c]，而我们直接读取文件格式是[n,c,h,w]，为了转成[n,h,w,c]形势，可以采用三种方法：

img = np.transpose(img, (0, 2, 3, 1))img = img.reshape(img.shape[0],img.shape[2],img.shape[3],img.shape[1])img_new = zeros((img.shape[0], img.shape[2],img.shape[3],img.shape[1]), dtype = np.float32)for c in range(0, img.shape[1]):for i in range(0, img.shape[2]):for j in range(0, img.shape[3]):img_new[:,c,i,j] = img_new[:,i,j,c]

有问题欢迎指出来，咩~~~

6月23号，又遇到了同样问题，百度找到之前写的解决方法，好记性不如烂笔头呀。。

问题背景：读取遥感数据[c, h, w]，但在keras的输入形式[h,w,c]，因此如要将[c, h, w]转化为[h,w,c]

上述的代码看着就头痛，不想看了，直接一句话：

img_hwc = np.transpose(img_chw, (1, 2,0))