2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed source of validation images (again)

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-17 13:48:12 +02:00
parent 0091027e1d
commit 2e388c0f82
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/data.py
View File

@ -106,7 +106,7 @@ def load_coco_val(data_path: str, category: int,
images = coco_val.loadImgs(img_ids) # load all images images = coco_val.loadImgs(img_ids) # load all images
annotation_ids = coco_val.getAnnIds(img_ids, catIds=[category]) annotation_ids = coco_val.getAnnIds(img_ids, catIds=[category])
annotations = coco_val.loadAnns(annotation_ids) # load all image annotations 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} ids_to_images = {image['id']: image for image in images}
checked_file_names, checked_bboxes = _clean_dataset(annotations, file_names, ids_to_images) checked_file_names, checked_bboxes = _clean_dataset(annotations, file_names, ids_to_images)