|
class StimulusSetLoader: |
|
def __init__(self, csv_path, stimuli_directory, cls): |
|
self.csv_path = csv_path |
|
self.stimuli_directory = stimuli_directory |
|
self.cls = cls |
|
|
|
def load(self): |
|
stimulus_set = pd.read_csv(self.csv_path) |
|
stimulus_set = StimulusSet(stimulus_set) |
|
stimulus_set.image_paths = {row['image_id']: os.path.join(self.stimuli_directory, row['filename']) |
|
for _, row in stimulus_set.iterrows()} |
|
assert all(os.path.isfile(image_path) for image_path in stimulus_set.image_paths.values()) |
|
return stimulus_set |
the class field is never used
