torchvision载入celeba的坑

最新推荐文章于 2024-02-09 18:19:25 发布

_Raymond_

最新推荐文章于 2024-02-09 18:19:25 发布

阅读量559

点赞数

文章标签：深度学习 celeba

本文链接：https://blog.csdn.net/weixin_43745072/article/details/130444039

版权

在使用torchvision加载CelebA数据集时，需注意不能直接使用香港中文大学提供的数据路径。应确保img_align_celeba目录下有Anno/和Eval中整理过的txt文件，以通过完整性检查。文章提供了预处理代码，包括Resize、ToTensor和Normalize步骤，用于训练(train)和验证(val)数据集。

摘要由CSDN通过智能技术生成

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

torchvision在检查完整性的时候。会算list_eval_partition.txt等txt文件。要从Anno/和Eval里面复制整理到img_align_celeba同级目录，然后
在这里插入图片描述

最后要摆成这个样子。
在这里插入图片描述
完整的代码，包括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)