plt.imshow(img.permute(1, 2, 0))
时间: 2024-05-31 22:09:06 浏览: 160
python中plt.imshow与cv2.imshow显示颜色问题
这段代码使用 PyTorch 中的张量表示图像,将其按照通道维度的顺序重新排列,并使用 Matplotlib 库中的 imshow 函数显示图像。其中,permute(1, 2, 0) 表示将原始张量的第 1 维移动到第 0 维,第 2 维移动到第 1 维,第 0 维移动到第 2 维,即从 (C, H, W) 变为 (H, W, C) 的形式。这样做是因为 Matplotlib 中使用的是通道在最后的图像格式。
阅读全文