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)