imgs = [img.permute(1, 2, 0) for img in imgs]
:
imgs
是一个包含多个图像张量的列表。img.permute(1, 2, 0)
是对每个图像张量进行维度排列。- 在 PyTorch 中,图像通常是
[C, H, W]
形状,即通道数(Channels)、高度(Height)、宽度(Width)。 permute(1, 2, 0)
将维度重新排列为[H, W, C]
,这是 Matplotlib 和其他图像处理库所期望的形状。
- 在 PyTorch 中,图像通常是
- 列表推导式
[img.permute(1, 2, 0) for img in imgs]
对imgs
列表中的每个图像进行维度排列转换。