一、原因 由于transforms.ToTensor()有维度转换的功能,图片经过转换后维度发生了变化,使得通道数在第一维,而plt.imshow()要求通道数在最后一维,所以要使用torch.permute()进行转换后,再转换成ndarray格式进行图片显示。 二、方法 imgs, labels = next(iter(data_dl)) img = imgs[0].permute(1, 2, 0).numpy() plt.imshow(img) plt.show()