Update extract_multilabel_pics.py

coco = COCO(

    '/softpro/ss18/caption/cocoapi/annotations/train/instances_train2017.json')

    '/../../cocoapi/annotations/train/instances_train2017.json')

cats = coco.loadCats(coco.getCatIds())

nms = [cat['name'] for cat in cats]

path = '/../../cocoapi/multilabel_outer/'

subfolder = path + 'multilabel_images/'

if not os.path.exists(path):

    print(f'Folder multilabel not existing.\nCreatue folder: {path}')

    os.mkdir(path)

    print(f'Create Subfolder: {subfolder}')

    os.mkdir(subfolder)

for category in nms:

    print(category)

    catIds = coco.getCatIds(catNms=[category])

    imgIds = coco.getImgIds(catIds=catIds)

    for pic in imgIds:

        name = coco.loadImgs(pic)[0]['file_name']

        shutil.copy2('images/train/' + name, 'multilabel_images/' + name)

        shutil.copy2('/../../cocoapi/images/train/' + name, subfolder + name)