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

Updated ssd_train function to use new signature of predict function

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-11 14:56:42 +02:00
parent 6c431e5e70
commit e8a82fbfaf
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -289,12 +289,13 @@ def _ssd_test(args: argparse.Namespace) -> None:
nr_digits = _get_nr_digits(length_dataset, batch_size) nr_digits = _get_nr_digits(length_dataset, batch_size)
steps_per_epoch = _get_nr_batches(length_dataset, batch_size) steps_per_epoch = _get_nr_batches(length_dataset, batch_size)
ssd.predict(test_generator, ssd.predict(test_generator,
steps_per_epoch,
ssd_model, ssd_model,
use_dropout, steps_per_epoch,
forward_passes_per_image,
image_size, image_size,
batch_size,
forward_passes_per_image,
output_path, output_path,
use_dropout,
nr_digits) nr_digits)