原始代码:
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)即可。