-
Notifications
You must be signed in to change notification settings - Fork 22
/
dataloader.py
24 lines (17 loc) · 821 Bytes
/
dataloader.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import torchvision.transforms as transforms
import torchvision.datasets as dset
import torchvision.utils as vutils
from PIL import ImageFile
import torch
ImageFile.LOAD_TRUNCATED_IMAGES = True
def loadImgs(des_dir = "./data/",img_size=128,batchSize = 20):
dataset = dset.ImageFolder(root=des_dir,
transform=transforms.Compose([
transforms.Resize(img_size),
transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
]))
dataloader = torch.utils.data.DataLoader(dataset,
batch_size= batchSize,
shuffle=True)
return dataloader