LEI_TTY7 2021-10-21 09:41 采纳率: 75%
浏览 551
已结题

Python读取文件的报错,该如何调整数据集位置

在学习案例:基于DCGAN的动漫人物生成时,已经将数据集和DCGAN.py放在同一文件夹下仍然出现报错
数据集是faces,一个存放了51,223个图片的文件夹
我的代码是

import torch
import torchvision
import torchvision.utils as vutils
import torch.nn as nn
from GanModel import Generator, Discriminator

# 设置超参数
BatchSize = 8
ImageSize = 96
Epoch = 25
Lr = 0.0002
Beta1 = 0.5
DataPath = './faces/'
OutPath = './imgs/'
# 定义是否使用GPU
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")


def train(netG, netD, dataloader):
    criterion = nn.BCELoss()
    optimizerG = torch.optim.Adam(netG.parameters(), lr=Lr, betas=(Beta1, 0.999))
    optimizerD = torch.optim.Adam(netD.parameters(), lr=Lr, betas=(Beta1, 0.999))

    label = torch.FloatTensor(BatchSize)
    real_label = 1
    fake_label = 0

    for epoch in range(1, Epoch + 1):
        for i, (imgs, _) in enumerate(dataloader):
            # 固定生成器G,训练鉴别器D
            optimizerD.zero_grad()
            # 让D尽可能的把真图片判别为1
            imgs = imgs.to(device)
            output = netD(imgs)
            label.data.fill_(real_label)
            label = label.to(device)
            errD_real = criterion(output, label)
            errD_real.backward()
            # 让D尽可能把假图片判别为0
            label.data.fill_(fake_label)
            noise = torch.randn(BatchSize, 100, 1, 1)
            noise = noise.to(device)
            fake = netG(noise)
            # 避免梯度传到G,因为G不用更新
            output = netD(fake.detach())
            errD_fake = criterion(output, label)
            errD_fake.backward()
            errD = errD_fake + errD_real
            optimizerD.step()

            # 固定鉴别器D,训练生成器G
            optimizerG.zero_grad()
            # 让D尽可能把G生成的假图判别为1
            label.data.fill_(real_label)
            label = label.to(device)
            output = netD(fake)
            errG = criterion(output, label)
            errG.backward()
            optimizerG.step()
            if i % 50 == 0:
                print('[%d/%d][%d/%d] Loss_D: %.3f Loss_G %.3f'
                      % (epoch, Epoch, i, len(dataloader), errD.item(), errG.item()))

        vutils.save_image(fake.data,
                          '%s/fake_samples_epoch_%03d.png' % (OutPath, epoch),
                          normalize=True)
        torch.save(netG.state_dict(), '%s/netG_%03d.pth' % (OutPath, epoch))
        torch.save(netD.state_dict(), '%s/netD_%03d.pth' % (OutPath, epoch))

if __name__ == "__main__":
    # 图像格式转化与归一化
    transforms = torchvision.transforms.Compose([
        torchvision.transforms.Scale(ImageSize),
        torchvision.transforms.ToTensor(),
        torchvision.transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
    dataset = torchvision.datasets.ImageFolder(DataPath, transform=transforms)

    dataloader = torch.utils.data.DataLoader(
        dataset=dataset,
        batch_size=BatchSize,
        shuffle=True,
        drop_last=True,
    )

    netG = Generator().to(device)
    netD = Discriminator().to(device)
    train(netG, netD, dataloader)

报错


```python

runfile('D:/DCGAN.py', wdir='D:')
Reloaded modules: GanModel
Traceback (most recent call last):

  File "D:\DCGAN.py", line 76, in <module>
    dataset = torchvision.datasets.ImageFolder(DataPath, transform=transforms)

  File "D:\Anaconda\lib\site-packages\torchvision\datasets\folder.py", line 310, in __init__
    super(ImageFolder, self).__init__(root, loader, IMG_EXTENSIONS if is_valid_file is None else None,

  File "D:\Anaconda\lib\site-packages\torchvision\datasets\folder.py", line 145, in __init__
    classes, class_to_idx = self.find_classes(self.root)

  File "D:\Anaconda\lib\site-packages\torchvision\datasets\folder.py", line 221, in find_classes
    return find_classes(directory)

  File "D:\Anaconda\lib\site-packages\torchvision\datasets\folder.py", line 42, in find_classes
    raise FileNotFoundError(f"Couldn't find any class folder in {directory}.")

FileNotFoundError: Couldn't find any class folder in ./faces/.

```

  • 写回答

2条回答 默认 最新

  • CSDN专家-kaily 2021-10-21 10:30
    关注

    DataPath = './faces/'
    OutPath = './imgs/'
    如果faces和当前py文件在同一个文件夹下,就直接去掉./,imgs同理
    如果还是不行,你就截个图让我们看一下你的当前文件夹目录或者结构是什么样的

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

问题事件

  • 已结题 (查看结题原因) 10月21日
  • 已采纳回答 10月21日
  • 创建了问题 10月21日

悬赏问题

  • ¥15 基于RTKLIB框架写的精密单点定位-AR
  • ¥15 PFENet的预训练权重
  • ¥15 程序哪有错误怎么改?
  • ¥15 交换机和交换机之间的链路带宽以及主机带宽的理解
  • ¥15 ai创想家对战模式代码
  • ¥15 集合A由3个2行4列二维数组构成,从集合A中任意取一个二维数组元素、如果该二维数组元素的对应列位置的上、下两数都是奇数,而且仅有2个列是奇数/奇数,则该数组有意义,并放入集合B中打印输出。
  • ¥15 电信IPV6 无法外网访问吗
  • ¥15 有偿求效果比较好的遥感影像匹配的c++代码
  • ¥15 博主,你好,我下载了你的智能网联汽车辅助驾驶安全信息检测系统,现在不会运行,可以教我吗,
  • ¥15 怎么在excle输入下列公式