2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Only use one trajectory for training and validation data each

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-04 13:15:51 +02:00
parent 7f5a32a441
commit bd499425ed
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/twomartens/masterthesis/cli.py
View File

@ -75,13 +75,15 @@ def _ssd_train(args: argparse.Namespace) -> None:
predictor_sizes=ssd_model.predictor_sizes, predictor_sizes=ssd_model.predictor_sizes,
batch_size=batch_size, batch_size=batch_size,
resized_shape=(image_size, image_size), resized_shape=(image_size, image_size),
training=True, evaluation=False, augment=False) training=True, evaluation=False, augment=False,
nr_trajectories=1)
val_generator, val_length = \ val_generator, val_length = \
data.load_scenenet_data(file_names_val, instances_val, args.coco_path, data.load_scenenet_data(file_names_val, instances_val, args.coco_path,
predictor_sizes=ssd_model.predictor_sizes, predictor_sizes=ssd_model.predictor_sizes,
batch_size=batch_size, batch_size=batch_size,
resized_shape=(image_size, image_size), resized_shape=(image_size, image_size),
training=False, evaluation=False, augment=False) training=False, evaluation=False, augment=False,
nr_trajectories=1)
del file_names_train, instances_train, file_names_val, instances_val del file_names_train, instances_train, file_names_val, instances_val
if args.debug: if args.debug: