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)
上一篇:如何使用光盘构建yum仓库


下一篇:2021-11-02