Loading preprocessing/coco_preprocessing/categories_of_pictures.py +6 −2 Original line number Diff line number Diff line Loading @@ -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] Loading Loading
preprocessing/coco_preprocessing/categories_of_pictures.py +6 −2 Original line number Diff line number Diff line Loading @@ -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] Loading