[DATA] Custom Image Data 넣기

파일 구조 

data

└resourve

    └happy

          └emoji00000.png

          └emoji00001.png

          └emoji00002.png

          └ ...

---

구조 확인하기

import glob

glob.glob('data/resource/happy/*')[:20]

---

config.augmentation = transforms.Compose([
                        transforms.Resize((config.img_shape[1], config.img_shape[2])), #resize
                        transforms.ToTensor(),
                        transforms.Normalize(mean=[0.5], std=[0.5]) #normalization
                      ])
                      

elif config.dataset == 'emoji': 
    train_dataset = datasets.ImageFolder(config.data_path,
                                       transform=config.augmentation
                                     )
                                     
train_loader = DataLoader(train_dataset, batch_size=config.batch_size, shuffle=True)
print(train_dataset)