From 6b56ba9aba08c7063b9d088048683c4599e76d27 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 4 Sep 2019 11:39:15 +0200 Subject: [PATCH] Changed annotation meta file to speed up saving images Signed-off-by: Jim Martens --- src/twomartens/masterthesis/debug.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/debug.py b/src/twomartens/masterthesis/debug.py index 8751d84..4291c82 100644 --- a/src/twomartens/masterthesis/debug.py +++ b/src/twomartens/masterthesis/debug.py @@ -51,7 +51,7 @@ def save_ssd_train_images(images: Union[np.ndarray, Sequence[str]], labels: np.n custom_string: optional custom string that is prepended to file names """ - annotation_file_train = f"{coco_path}/annotations/instances_train2014.json" + annotation_file_train = f"{coco_path}/annotations/instances_minival2014.json" _, _, _, classes_to_names = get_coco_cat_maps_func(annotation_file_train) colors = pyplot.cm.hsv(np.linspace(0, 1, 61)).tolist() os.makedirs(output_path, exist_ok=True)