permute()函数功能

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 和其他图像处理库所期望的形状。
  • 列表推导式 [img.permute(1, 2, 0) for img in imgs]imgs 列表中的每个图像进行维度排列转换。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值