проблема с загрузкой данных в Pytorch в локальном блокноте Jupyter - PullRequest
0 голосов
/ 01 февраля 2020

Я пытался использовать 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
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...