From 41cf8074bd145fdd8f066b7db52c92dec6cb4fa2 Mon Sep 17 00:00:00 2001 From: HombreLaser Date: Mon, 11 Sep 2023 17:55:46 -0600 Subject: Fix the dataset utils class --- dataset_utils.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) (limited to 'dataset_utils.py') diff --git a/dataset_utils.py b/dataset_utils.py index 865d5be..fda75d2 100644 --- a/dataset_utils.py +++ b/dataset_utils.py @@ -6,23 +6,21 @@ from pathlib import Path class DatasetUtils: def __init__(self) -> None: self.data_path = Path('./data') - # self.current_directory = 'I' """Convert the dataset to a 2 dimension array.""" def data_to_arrays(self): for dir in self.data_path.iterdir(): if not dir.is_dir(): continue - for file in dir.glob('/*.png'): + for file in dir.glob('*.png'): image = Image.open(str(file)).convert(mode='L') # Return the image's pixel values as an array alongside # the character that it represents. - yield (dir.name, numpy.asarray(image)) + yield (dir.name, numpy.asarray(image).ravel()) """Search for a file in the dataset.""" def search(self, filename: str): - dataset = [image for image in self.data_path.rglob('*.png')] - + pass def get_random_sample(self): pass -- cgit v1.2.3