Machine Learning/Model

[DATA] Custom Image Data 넣기

파송송 2022. 12. 28. 20:03
728x90

파일 구조 

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)

 

728x90