from pathlib import Path

from dataset.dataset import Dataset

train_dataset = Dataset(Path('data/resized_dataset/train'))
valid_dataset = Dataset(Path('data/resized_dataset/valid'))

for i in train_dataset.take(1):
    print(i)