11 lines
230 B
Python
11 lines
230 B
Python
|
|
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)
|