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

image_recognition #5

Merged
s473558 merged 23 commits from image_recognition into master 2023-06-12 16:25:59 +02:00
Conversation 0 Commits 23 Files Changed 15 +2 -2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit da73e223e3 - Show all commits

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_loader = DataLoader(
train_dataset, batch_size=BATCH_SIZE, shuffle=True, num_workers=0, pin_memory=True train_dataset, batch_size=BATCH_SIZE, shuffle=True, num_workers=0, pin_memory=True