def main(): """ Entry point when using CRNN from the commandline """ args = parse_arguments() if not args.train and not args.test: print("If we are not training, and not testing, what is the point?") crnn = None if args.train: crnn = CRNN( args.batch_size, args.model_path, args.examples_path, args.max_text_length, args.max_image_width, args.train_test_ratio ) crnn.train(args.iteration_count) crnn.save() if args.test: if crnn is None: crnn = CRNN( args.batch_size, args.model_path, args.examples_path, args.max_text_length, args.max_image_width, 1 ) crnn.test()
def start_train(): model = CRNN(MODEL_HYPER.batch_size, MODEL_HYPER.epoches, MODEL_HYPER.data_path, MODEL_HYPER.text_path, MODEL_HYPER.log_path, MODEL_HYPER.model_path) model.train() model.save()