File size: 161 Bytes
eecd883
 
 
 
 
1
2
3
4
5
from torchvision.datasets import ImageFolder
from transformation import transforms

data_path = 'Data'
dataset = ImageFolder(data_path, transform=transforms)