added categories_labels.py

from pycocotools.coco import COCO

import numpy as np

import skimage.io as io

import matplotlib.pyplot as plt

import pylab

import json

import os

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

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

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

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

with open('labels.txt', 'w') as handle:

    for category in nms:

        out = category + '\n'

        handle.write(out)