admin 管理员组

文章数量: 887039

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

原始代码:

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)即可。

本文标签: TypeError Cannot handle this data type (1 1 128)