2020-11-19

为了保存二值化的mask，将pytorch的tensor转成numpy，然后转成Image类进行保存。

但总是出现cannot handle this data type

查阅良久，发现Image.fromarray函数只能处理彩色（H, W, 3）或者灰度图(H, W)。因此要将灰度图的（256， 256， 1）reshape到（256， 256）

save_path = os.path.join(opt.save_root, img_dir.replace('.png', '_{}.png'.format(str(i))))
img_save = (predicte_mask[idx] * 255.0).clamp(0, 255).to(torch.uint8).permute(1, 2, 0).reshape(256, 256)
util.save_image(img_save.cpu().numpy(), save_path)