Enabled augmentations for training again

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-07-22 15:20:46 +02:00
parent d1dadabc0d
commit 2b656057af

View File

@ -774,7 +774,7 @@ def _ssd_train_get_generators(args: argparse.Namespace,
predictor_sizes=predictor_sizes, predictor_sizes=predictor_sizes,
batch_size=batch_size, batch_size=batch_size,
image_size=image_size, image_size=image_size,
training=True, evaluation=False, augment=False, training=True, evaluation=False, augment=True,
debug=args.debug, debug=args.debug,
nr_trajectories=nr_trajectories) nr_trajectories=nr_trajectories)