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()
