Commit 5ec70957 authored by holzinger's avatar holzinger
Browse files

Update 100.py

parent 16e05713

preprocessing/coco_preprocessing/100.py

+3 −3
Original line number Diff line number Diff line
@@ -8,8 +8,8 @@ import pickle 
import shutil





path = '/softpro/ss18/caption/cocoapi/100/'

coco = COCO('/softpro/ss18/caption/cocoapi/annotations/train/instances_train2017.json')

path = '../../cocoapi/100/'

coco = COCO('../../cocoapi/annotations/train/instances_train2017.json')

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

nms = [cat['name'] for cat in cats]
@@ -29,5 +29,5 @@ for category in nms: 
        pic100[file_name] = (category)

        i += 1



with open('100_pics_or_didnt_happen.pickle', 'wb') as handle:

with open('../../cocoapi/100_pics_or_didnt_happen.pickle', 'wb') as handle:

    pickle.dump(pic100, handle, protocol=pickle.HIGHEST_PROTOCOL)