【问题标题】:How to reshape an Image in pytorch如何在pytorch中重塑图像
【发布时间】:2020-06-25 13:24:20
【问题描述】:

我有一个形状为(32, 3, 32, 32) 的图像。我知道它的形式是(batch_size, Channel, Height, Width)

问。如何将其转换为覆盖Channel(32, 32, 32)

【问题讨论】:

  • “覆盖通道”是什么意思?如果您有两个数字,您想如何将它们转换为一个数字?选最大的?取平均值?还是?
  • 我已将图像转换为灰度,我希望得到32, 32, 32 的输出而不是32, 1, 32, 32
  • 然后使用torch.squeeze压缩输出(即删除大小为1的维度)。
  • 谢谢!解决了

标签: python image-processing dataset pytorch


【解决方案1】:

如果你想转换成灰度,你可以这样做:

image.mean(dim=1)

【讨论】:

    猜你喜欢
    • 2021-07-06
    • 2018-10-27
    • 1970-01-01
    • 2023-02-14
    • 1970-01-01
    • 2015-08-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多