学习GAN中遇到的问题及解决

首先放一下我自己改完后的代码https://github.com/cui-jinming/pytorch-GAN

在GitHub上下载了一个pytorch版本的代码，打开以后完全看不懂，在b站找了个教学视频https://www.bilibili.com/video/av62138405

学完了开始改代码，原代码使用的是MNIST数据集，我想使用自己的数据集训练GAN

选择DCGAN模型改代码。

第一个问题，怎么使用自己的数据集：重写dataset，使用dataloader

DCGAN代码改好了，选择CGAN继续改

第一个问题，数据集对应的标签怎么改：使用torchvision.datasets.ImageFolder，直接根据文件夹名称就会变成文件的标签

第二个问题，报错：TypeError: 'list' object is not callable

解决办法：https://stackoverflow.com/questions/54431671/pytorch-dataloader-list-is-not-callable-error-when-enumerating

1 train_transforms = [
2     transforms.RandomResizedCrop(224),
3     transforms.RandomHorizontalFlip(),
4     transforms.ToTensor(),
5     transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])]

改成

1 train_transforms = transforms.Compose([
2     transforms.RandomResizedCrop(224),
3     transforms.RandomHorizontalFlip(),
4     transforms.ToTensor(),
5     transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])

posted on 2020-03-06 14:03 小明明明不是企鹅阅读(552) 评论(0) 编辑收藏举报