Pytorch以单通道(灰度图)加载图片

设置加载数据集时的transform参数如下即可：

from torchvision import datasets, transforms
transform = transforms.Compose(
    [

        transforms.Grayscale(num_output_channels=1),
        transforms.ToTensor()
    ]
) 
data = datasets.CIFAR10(root=".", download=True,transform=transform)

标签：datasets,transform,Pytorch,灰度,transforms,加载,单通道
来源： https://www.cnblogs.com/orion-orion/p/15862503.html