def _main(): tk.better_exceptions() parser = argparse.ArgumentParser() parser.add_argument('mode', choices=('check', 'train', 'fine', 'validate', 'predict')) parser.add_argument('--cv-index', default=0, choices=range(CV_COUNT), type=int) args = parser.parse_args() with tk.dl.session(use_horovod=args.mode in ('train', 'fine')): if args.mode == 'check': _create_network()[0].summary() elif args.mode == 'train': tk.log.init(MODELS_DIR / f'train.fold{args.cv_index}.log') _train(args) elif args.mode == 'fine': tk.log.init(MODELS_DIR / f'fine.fold{args.cv_index}.log') _train(args, fine=True) elif args.mode == 'validate': tk.log.init(MODELS_DIR / 'validate.log') _validate() elif args.mode == 'predict': tk.log.init(MODELS_DIR / 'predict.log') _predict()
def _main(): tk.better_exceptions() parser = argparse.ArgumentParser() parser.add_argument('--vocdevkit-dir', default=pathlib.Path('pytoolkit/data/VOCdevkit'), type=pathlib.Path) parser.add_argument('--result-dir', default=pathlib.Path('results'), type=pathlib.Path) parser.add_argument('--input-size', default=(320, 320), type=int, nargs=2) parser.add_argument('--batch-size', default=16, type=int) args = parser.parse_args() with tk.dl.session(): tk.log.init(args.result_dir / 'validate.log') _run(args)
def _main(): tk.better_exceptions() parser = argparse.ArgumentParser() parser.add_argument('mode', choices=('validate', 'predict', 'all'), nargs='?', default='all') args = parser.parse_args() if args.mode == 'validate': tk.log.init(MODELS_DIR / 'validate.log') _validate() elif args.mode == 'predict': tk.log.init(None) _predict() else: tk.log.init(MODELS_DIR / 'validate.log') _validate() _predict()