From 289275e82f9427fd8bad99cd9f09a3bb8c10510f Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 15 Jul 2019 13:34:54 +0200 Subject: [PATCH] Changed _ssd_evaluate_save_images to save all images Will need to be changed again if more than one trajectory is evaluated Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 16 +--------------- 1 file changed, 1 insertion(+), 15 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 67174f4..53ac9d4 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -369,22 +369,8 @@ def _ssd_evaluate_save_images(filenames: Sequence[str], labels: Sequence[np.ndar get_coco_cat_maps_func: callable, save_images: callable, image_size: int, batch_size: int, output_path: str, coco_path: str) -> None: - from PIL import Image - images = [] - for i, filename in enumerate(filenames): - if i == batch_size: - break - with Image.open(filename) as image: - images.append(np.array(image, dtype=np.uint8)) - - cleaned_labels = [] - for j, label in enumerate(labels): - if j == batch_size: - break - cleaned_labels.append(label) - - save_images(images, cleaned_labels, output_path, coco_path, image_size, get_coco_cat_maps_func) + save_images(filenames, labels, output_path, coco_path, image_size, get_coco_cat_maps_func) def _init_eager_mode() -> None: