【发布时间】: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