TypeError: Cannot handle this data type: (1, 1, 128), |u1

原始代码：

scipy.misc.toimage(img).save(s, format="png")

因为scipy1.2.0以上没有了.toimage等很多函数。所以运用其他的来代替。

from PIL import Image
Image.fromarray(np.uint8(img).convert('RGB').save(s, format="png")

但是运行会报错误：

TypeError: Cannot handle this data type: (1, 1, 128), |u1

解决措施：

Image.fromarray(np.uint8(img.transpose(1, 2, 0))).convert('RGB').save(s, format="png")

添加img.transpose(1,2,0)。

原因：我的数据输入格式img是（3,128,128），即是（channel，width，height）。但是使用PIL库处理图像数据时候，需要的格式是（width,height,channel），所以需要进行转换，把（C,W,H）变为（W,H,C）即可。