def main(): test_dataset = dataset.Dataset(base_path=FLAGS.base_dir, data_list_path=FLAGS.data_list_path) inference(test_dataset)
def main(args): test_dataset = dataset.Dataset(base_path=args.base_dir, data_list_path=args.data_list_path) inference(test_dataset)
def main(args): train_dataset = dataset.Dataset(base_path=args.base_dir, data_list_path=args.data_list_path) train(train_dataset)
def main(): train_dataset = dataset.Dataset(base_path=FLAGS.base_dir, data_list_path=FLAGS.data_list_path) train(train_dataset)