Commit 1e56fc18 authored by holzinger's avatar holzinger
Browse files

Update categories_of_pictures.py

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


path = '/../../cocoapi/images_txt/'
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')
coco = COCO('../../cocoapi/annotations/train/instances_train2017.json')
cats = coco.loadCats(coco.getCatIds())
nms = [cat['name'] for cat in cats]