From c2be08e0c8b570d32cbcc38ff1f6ee9e5ca05f34 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 11 Jul 2019 17:13:17 +0200 Subject: [PATCH] Changed code to save images with predictions not ground truth Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 3406db3..3f6d09d 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -325,7 +325,7 @@ def _ssd_evaluate(args: argparse.Namespace) -> None: predictions = _ssd_evaluate_unbatch_list(predictions_glob_string) _pickle(predictions_file, predictions) - _ssd_evaluate_save_images(filenames, labels, + _ssd_evaluate_save_images(filenames, predictions, coco_utils.get_coco_category_maps, debug.save_ssd_train_images, image_size, output_path, coco_path)