From 588e66c7723b9cd31e0e70f5cfe5552434d9a45d Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 11 Aug 2019 15:36:39 +0200 Subject: [PATCH] Only the first batch size images are saved on evaluate 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 f64c178..96d698d 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -386,7 +386,7 @@ def _ssd_evaluate_save_images(filenames: Sequence[str], labels: Sequence[np.ndar image_size: int, batch_size: int, output_path: str, coco_path: str) -> None: - save_images(filenames, labels, output_path, coco_path, image_size, get_coco_cat_maps_func) + save_images(filenames[:batch_size], labels[:batch_size], output_path, coco_path, image_size, get_coco_cat_maps_func) def _visualise_gt(args: argparse.Namespace,