Enabled augmentations for training again

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-22 15:20:46 +02:00
parent d1dadabc0d
commit 2b656057af
1 changed files with 1 additions and 1 deletions
--- a/src/twomartens/masterthesis/cli.py
+++ b/src/twomartens/masterthesis/cli.py
@ -774,7 +774,7 @@ def _ssd_train_get_generators(args: argparse.Namespace,
                  predictor_sizes=predictor_sizes,
                  batch_size=batch_size,
                  image_size=image_size,
-                  training=True, evaluation=False, augment=False,
+                  training=True, evaluation=False, augment=True,
                  debug=args.debug,
                  nr_trajectories=nr_trajectories)