def load_samples(self, directories):
files = []
for directory in directories:
files += helpers.get_file_names("../" + directory)
image_paths = [
file for file in files
if file[-4:] == ".jpg" or file[-4:] == ".png"
]
samples = []
for path in image_paths:
sample_id = path.split("/")[-1].split("_")[0]
sample_index = self.get_sample_index(samples, sample_id)
if sample_index == -1:
sample = Sample(sample_id)
sample.add_image(path)
samples.append(sample)
else:
samples[sample_index].add_image(path)
print("Samples found:")
[print("\t" + sample.get_id()) for sample in samples]
return samples
