图像通道转换[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]
有问题欢迎指出来，咩~~~
2021年6月23号，又遇到了同样问题，百度找到之前写的解决方法，好记性不如烂笔头呀。。
问题背景：读取遥感数据[c, h, w]，但在keras的输入形式[h,w,c]，因此如要将[c, h, w]转化为[h,w,c]
上述的代码看着就头痛，不想看了，直接一句话：
img_hwc = np.transpose(img_chw, (1, 2,0))