def main(): train_data_preprocessor = DataPreprocessor(dataset='train') test_data_preprocessor = DataPreprocessor(dataset='test') train_data_preprocessor.preprocess() test_data_preprocessor.preprocess() cnn = CNN() cnn.init_model() cnn.compile_model() cnn.train_model(train_data_preprocessor) cnn.evaluate_model(test_data_preprocessor) print('done')
def train_top_layers(cls): top_layers = CNN.get_top_layers(input_shape=CNN.input_shape()) top_layers_compiled = CNN.compile_model(top_layers) cls.fit_model(top_layers_compiled, save_full=False)