2024-05-05 01:20:04 +02:00
|
|
|
|
|
|
|
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'))
|
|
|
|
|
2024-05-05 13:25:53 +02:00
|
|
|
for i in train_dataset.take(1):
|
|
|
|
print(i)
|