torchvision.utils.save_image保存黑白灰度图结果出现灰色
原本像保存成的样子是黑白的灰度图,维度是(40,1,64,64),保存代码是:
torchvision.utils.save_image(real_data, path, cmap='gray')
![在这里插入图片描述](https://img-blog.csdnimg.cn/direct/3491a86150d24d53a0d61ed660445a2b.png)
结果是:
![在这里插入图片描述](https://img-blog.csdnimg.cn/direct/030c491000134a3db0b240b4a77c10d1.png)
后来发现少了一个normalize参数,应该使用的保存代码是:
torchvision.utils.save_image(real_data, path, 'real_data.png'), normalize=True,cmap='gray')