CSVDataLoader¶
-
class
csv_data_loader.
CSVDataLoader
(target_emotion_map, datapath=None, validation_split=0.2, image_dimensions=None, csv_label_col=None, csv_image_col=None, out_channels=1)[source]¶ DataLoader subclass loads image and label data from csv file.
Parameters: - emotion_map – Dict of target emotion label values and their corresponding label vector index values.
- datapath – Location of image dataset.
- validation_split – Float percentage of data to use as validation set.
- image_dimensions – Dimensions of sample images (height, width).
- csv_label_col – Index of label value column in csv.
- csv_image_col – Index of image column in csv.
- out_channels – Number of image channels.