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

Updated image paths to work with cvpc7 installation

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-15 10:21:36 +02:00
parent c8e49bd0da
commit 3afb74587b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -51,7 +51,7 @@ def load_coco(data_path: str, category: int,
images = coco_train.loadImgs(img_ids) # load all images images = coco_train.loadImgs(img_ids) # load all images
annotation_ids = coco_train.getAnnIds(img_ids, catIds=[category]) annotation_ids = coco_train.getAnnIds(img_ids, catIds=[category])
annotations = coco_train.loadAnns(annotation_ids) # load all image annotations annotations = coco_train.loadAnns(annotation_ids) # load all image annotations
file_names = {image['id']: f"{data_path}/images/train2014/{image['file_name']}" for image in images} file_names = {image['id']: f"{data_path}/train2014/{image['file_name']}" for image in images}
# load validation images # load validation images
coco_val = coco.COCO(annotation_file_val) coco_val = coco.COCO(annotation_file_val)
@ -59,7 +59,7 @@ def load_coco(data_path: str, category: int,
images_val = coco_val.loadImgs(img_ids) # load all images images_val = 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_val = coco_val.loadAnns(annotation_ids) # load all image annotations annotations_val = coco_val.loadAnns(annotation_ids) # load all image annotations
file_names_val = {image['id']: f"{data_path}/images/val2014/{image['file_name']}" for image in images_val} file_names_val = {image['id']: f"{data_path}/val2014/{image['file_name']}" for image in images_val}
images.extend(images_val) images.extend(images_val)
annotations.extend(annotations_val) annotations.extend(annotations_val)