From 2e388c0f8226392b8373ec90308adde2ec0a2250 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 17 Apr 2019 13:48:12 +0200 Subject: [PATCH] Fixed source of validation images (again) Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index 6509f66..5904f9a 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -106,7 +106,7 @@ def load_coco_val(data_path: str, category: int, images = coco_val.loadImgs(img_ids) # load all images annotation_ids = coco_val.getAnnIds(img_ids, catIds=[category]) annotations = coco_val.loadAnns(annotation_ids) # load all image annotations - file_names = {image['id']: f"{data_path}/minival2014/{image['file_name']}" for image in images} + file_names = {image['id']: f"{data_path}/val2014/{image['file_name']}" for image in images} ids_to_images = {image['id']: image for image in images} checked_file_names, checked_bboxes = _clean_dataset(annotations, file_names, ids_to_images)