Example #1
0
    def define_inputs_and_labels(self, args):

        self.input_shape = model_factory.define_input_shape(args)

        if self.action == 'interactive':
            self.reader = DataReader.InteractiveDataReader(self.input_shape[0], self.input_shape[1], args)
            self.inputs = self.reader.build_inputs()
        else:
            self.reader = DataReader.TrainDataReader(self.input_shape, args)
            self.inputs, self.labels, self.filenames = self.reader.build_iterator()
        self.classnames = self.reader.classnames
        self.nclasses = len(self.classnames)