Я пытался использовать 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