Use one trajectory only for prediction

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-07-09 12:03:38 +02:00
parent 0c7e96ee9d
commit 1d372aba59

View File

@ -341,7 +341,8 @@ def _ssd_test(args: argparse.Namespace) -> None:
predictor_sizes=ssd_model.predictor_sizes,
batch_size=batch_size,
resized_shape=(image_size, image_size),
training=False, evaluation=True, augment=False)
training=False, evaluation=True, augment=False,
nr_trajectories=1)
del file_names_photos, instances
nr_digits = math.ceil(math.log10(math.ceil(length_dataset / batch_size)))