s473558/si23traktor
2
2
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add more code flexibility

Browse Source
This commit is contained in:
s473558 2023-06-05 04:48:04 +02:00
parent e27acbacaf
commit da73e223e3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
neural_network/datasets.py
View File

@ -29,9 +29,9 @@ valid_transform = transforms.Compose([
)
])
train_dataset = torchvision.datasets.ImageFolder(root='./Vegetable Images/train', transform=train_transform)
train_dataset = torchvision.datasets.ImageFolder(root='./images/train', transform=train_transform)
validation_dataset = torchvision.datasets.ImageFolder(root='./Vegetable Images/validation', transform=valid_transform)
validation_dataset = torchvision.datasets.ImageFolder(root='./images/validation', transform=valid_transform)
train_loader = DataLoader(
train_dataset, batch_size=BATCH_SIZE, shuffle=True, num_workers=0, pin_memory=True