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)