在本地Jupyter Notebook的Pytorch中上传数据时遇到问题

时间:2020-02-01 08:28:21

标签: jupyter-notebook pytorch

我尝试使用Torchvision,但是我不知道加载本地数据的代码。数据文件不是CSV格式。它已经分解为训练和测试。

这是我输入的代码

dataset = datasets.ImageFolder(root=dir,transform = transform)

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

dir='G:/Cat_Dog_data/train'

这是错误

    TypeError                                 Traceback (most recent call last)
    <ipython-input-34-ae805628f682> in <module>
    ----> 1 dataset = datasets.ImageFolder(root=dir,transform = transform)
          2 
          3 dataloader = torch.utils.data.DataLoader(dataset,batch_size=32, shuffle =True)
          4 
          5 dir='G:/Cat_Dog_data/train'

    ~\Anaconda3\lib\site-packages\torchvision\datasets\folder.py in __init__(self, root, transform, target_transform, loader, is_valid_file)
        207                                           transform=transform,
        208                                           target_transform=target_transform,
    --> 209                                           is_valid_file=is_valid_file)
        210         self.imgs = self.samples

    ~\Anaconda3\lib\site-packages\torchvision\datasets\folder.py in __init__(self, root, loader, extensions, transform, target_transform, is_valid_file)
         91         super(DatasetFolder, self).__init__(root, transform=transform,
         92                                             target_transform=target_transform)
    ---> 93         classes, class_to_idx = self._find_classes(self.root)
         94         samples = make_dataset(self.root, class_to_idx, extensions, is_valid_file)
         95         if len(samples) == 0:

    ~\Anaconda3\lib\site-packages\torchvision\datasets\folder.py in _find_classes(self, dir)
        120         if sys.version_info >= (3, 5):
        121             # Faster and available in Python 3.5 and above
    --> 122             classes = [d.name for d in os.scandir(dir) if d.is_dir()]
        123         else:
        124             classes = [d for d in os.listdir(dir) if os.path.isdir(os.path.join(dir, d))]

    TypeError: scandir: path should be string, bytes, os.PathLike or None, not builtin_function_or_method

0 个答案:

没有答案
相关问题