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