torchvision载入celeba的坑

celeba载入数据集的时候,一定要注意不能直接把香港中文大学分享的数据路径摆进去。

torchvision在检查完整性的时候。会算list_eval_partition.txttxt文件。要从Anno/Eval里面复制整理到img_align_celeba同级目录,然后
torchvision载入celeba的坑_第1张图片

最后要摆成这个样子。
torchvision载入celeba的坑_第2张图片
完整的代码,包括mean和std

tfms = transforms.Compose([transforms.Resize((256, 256)),
                           transforms.ToTensor(),
                           transforms.Normalize(mean=(0.50612009, 0.42543493, 0.38282761),std=(0.26589054, 0.24521921, 0.24127836))
                        ])
train_datasets=torchvision.datasets.CelebA(root="CelebA",split="train",download=False,transform=tfms)
val_datasets=torchvision.datasets.CelebA(root="CelebA",split="valid",download=False,transform=tfms)
    

你可能感兴趣的:(深度学习,celeba)