Pytorch：加载灰度图像的数据集

Question

我想加载灰度图像的数据集。我使用了ImageFolder，但是默认情况下不会加载灰度图像，因为它将图像转换为RGB。

[我找到了使用ImageFolder加载图像并在使用以下方法将图像转换为灰度后的解决方案：

transforms.Grayscale(num_output_channels=1)

或

ImageOps.grayscale(image)

正确吗？如何加载未经转换的灰度图像？我尝试使用ImageDataBunch，但是导入fastai.vision]时遇到问题

我想加载灰度图像的数据集。我使用了ImageFolder，但是默认情况下不会加载灰度图像，因为它将图像转换为RGB。我找到了使用ImageFolder和...

Answer 1

是，这是正确的，默认情况下AFAIK pillow将图像加载到RGB中，请参见例如answers to this question。因此，转换为grayscale是唯一的方法，尽管当然需要时间。