Pytorch 加载单通道图片用作数据集训练和测试

用transforms.Grayscale(1)来实现。如下所示：
train_dataset = datasets.ImageFolder(
    traindir,
    transforms.Compose([
        #transforms.RandomResizedCrop(224),
        transforms.Resize((112,112)),
        transforms.RandomHorizontalFlip(),
        transforms.Grayscale(1),  ############
        transforms.ToTensor(),
    ]))

train_sampler = None

train_loader = torch.utils.data.DataLoader(
    train_dataset, batch_size=args.batch_size, shuffle=True,
    num_workers=args.workers, pin_memory=True, sampler=train_sampler)

val_loader = torch.utils.data.DataLoader(
    datasets.ImageFolder(valdir, transforms.Compose([
        transforms.Resize((112,112)),
        #transforms.CenterCrop(224),
        transforms.Grayscale(1), ############
        transforms.ToTensor(),
    ])),
    batch_size=args.batch_size, shuffle=False,
    num_workers=args.workers, pin_memory=True)