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