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.
load_data()[source]

Loads image and label data from specified csv file path.

Returns:Dataset object containing image and label data.