// read each line of csv files into an instance of Subject
if self.is_training:
self.readers = [ImageReader(SUPPORTED_INPUT, "train")]
if self.has_validation_data and self.action_param.validate_every_n:
self.readers.append(ImageReader(SUPPORTED_INPUT, "validation"))
else: // in the inference process use image input only
self.readers = [ImageReader(["image"], "inference")]
After Change
// read each line of csv files into an instance of Subject
if self.is_training:
reader_train = ImageReader(SUPPORTED_INPUT)
file_list = data_partitioner.get_file_list(TRAIN)
reader_train.initialise_reader(data_param, task_param, file_list)
self.readers = [reader_train]
if self.has_validation_data and self.action_param.validate_every_n: