|
|
|
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)
|