PyTorch 报错：TypeError: Cannot handle this data type: (1, 1, 512), |u1 （已解决）

PyTorch 报错：TypeError: Cannot handle this data type: (1, 1, 512), |u1 （已解决）

 

pytorch 代码，保存图片

语句时，报错

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

 

这是因为，当要保存的图片为 灰度图像 时，灰度图像的 numpy 尺度是 [1, h, w]；这就会报错。

需要将 [1, h, w] 改变为 [h, w]， 所以，在将 tensor 格式转为 numpy 时，要将维度变过来：

这里，当 image_numpy.shape[0]=1，即 image_numpy 是灰度图像，只要将 image_numpy = image_numpy[0]，其格式就从 [1, h, w] 改变为 [h, w] 了。