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

Use one trajectory only for prediction

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-09 12:03:38 +02:00
parent 0c7e96ee9d
commit 1d372aba59
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -341,7 +341,8 @@ def _ssd_test(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=False, evaluation=True, augment=False) training=False, evaluation=True, augment=False,
nr_trajectories=1)
del file_names_photos, instances del file_names_photos, instances
nr_digits = math.ceil(math.log10(math.ceil(length_dataset / batch_size))) nr_digits = math.ceil(math.log10(math.ceil(length_dataset / batch_size)))