Commit 90f6582d authored by holzinger's avatar holzinger
Browse files

Update categories_of_pictures.py

parent d40419a3
Loading
Loading
Loading
Loading
+6 −2
Original line number Diff line number Diff line
@@ -8,8 +8,12 @@ import json
import os


path = '/softpro/ss18/caption/cocoapi/images_txt/'
coco = COCO('/softpro/ss18/caption/cocoapi/annotations/train/instances_train2017.json')
path = '/../../cocoapi/images_txt/'
if not os.path.exists(path):
    print(f'Folder images_txt for image labels not existing.\nCreatue folder: {path}')
    os.mkdir(path)

coco = COCO('/../../cocoapi/annotations/train/instances_train2017.json')
cats = coco.loadCats(coco.getCatIds())
nms = [cat['name'] for cat in cats]