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 image, labels in train_dataset.dataset.take(1):
    print(image, labels)