如何将形状(批次、通道、高度、宽度)的图像的pytorch张量转换为(批次、高度、宽)

Question

我想通过移除颜色通道将彩色图像转换为灰度图像。这可能吗？

Answer 1

您可以使用Grayscale转换

https://pytorch.org/vision/stable/generated/torchvision.transforms.Grayscale.html

from torchvision.transforms import Grayscale
grayscale_batch = Grayscale()(color_batch)

这产生了一个(batch_size，1，H，W(张量。

要删除通道尺寸，请应用torch.squeeze()